Home
last modified time | relevance | path

Searched defs:operand_shape (Results 1 – 25 of 49) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dshape_inference.cc433 const Shape& operand_shape, PrimitiveType new_element_type) { in InferConvertShape()
449 const Shape& operand_shape, PrimitiveType new_element_type) { in InferBitcastConvertShape()
503 const Shape& operand_shape, const int exponent_bits, in InferReducePrecisionShape()
527 const Shape& operand_shape, const Shape& padding_value_shape, in InferPadShape()
1210 const Shape& operand_shape, const Shape& scale_shape, in InferBatchNormTrainingShape()
1315 const Shape& operand_shape, const Shape& scale_shape, in InferBatchNormInferenceShape()
1459 const Shape& operand_shape, const Shape& scale_shape, in InferBatchNormGradShape()
2043 for (const Shape* operand_shape : operand_shapes) { in InferAllGatherShape() local
2081 for (const Shape* operand_shape : operand_shapes) { in InferAllReduceShape() local
2099 for (const Shape* operand_shape : operand_shapes) { in InferReduceScatterShape() local
[all …]
H A Dhlo_verifier.cc1010 const Shape& operand_shape = instruction.operands()[i]->shape(); in SameElementTypesForOperandsAndToApplyParameters() local
1046 const Shape& operand_shape = bitcast->operand(0)->shape(); in HandleBitcast() local
1069 const Shape& operand_shape = broadcast->operand(0)->shape(); in HandleBroadcast() local
1087 const Shape& operand_shape = dynamic_reshape->operand(0)->shape(); in HandleDynamicReshape() local
1101 const Shape& operand_shape = reshape->operand(0)->shape(); in HandleReshape() local
1510 const Shape& operand_shape = copy_done->operand(0)->shape(); in HandleCopyDone() local
2319 const Shape& operand_shape = operand->shape(); in CheckElementwiseInstruction() local
2331 const Shape& operand_shape = comparison->operand(1)->shape(); in CheckElementwiseInstruction() local
2517 const Shape& operand_shape = operand->shape(); in Postprocess() local
H A Dlogistic_expander.cc43 const Shape operand_shape = operand->shape(); in ExpandLogisticWithTanh() local
59 const Shape operand_shape = operand->shape(); in ExpandLogisticWithExp() local
H A Dhlo_creation_utils.cc560 const Shape& operand_shape = operand->shape(); in CollapseFirstNDims() local
585 const Shape& operand_shape = operand->shape(); in PrependDegenerateDims() local
623 const Shape& operand_shape = operand->shape(); in InsertDegenerateDims() local
H A Dhlo_sharding_util.cc909 const Shape& operand_shape, const HloSharding& operand_sharding, in PassthroughOperandToGatherOutputOrScatterUpdate()
956 const Shape& operand_shape, const HloSharding& update_or_gather_sharding, in PassthroughGatherOutputOrScatterUpdateToOperand()
1064 const Shape& operand_shape) { in GatherOutputShardingFromDataOperand()
1118 std::vector<int64_t> GetScatterSliceSize(const Shape& operand_shape, in GetScatterSliceSize()
H A Dbatchnorm_expander.cc172 const Shape operand_shape = operand->shape(); in HandleBatchNormTraining() local
301 const Shape operand_shape = operand->shape(); in HandleBatchNormInference() local
H A Dreduce_decomposer.cc141 auto operand_shape = operand->shape(); in ExpectedOutputShape() local
H A Dindexed_array_analysis.cc351 absl::Span<const int64_t> operand_shape, in ComputeReshapePassthroughDimPairs()
450 absl::Span<const int64_t> operand_shape, in FindSourcePositionForPassthroughResultDim()
H A Ddynamic_padder.cc475 const Shape operand_shape = reshape->operand(0)->shape(); in RewriteDynamicReshapeSplitInput() local
779 const Shape operand_shape = reshape->operand(0)->shape(); in RewriteDynamicReshapeSingleGroup() local
1361 Shape operand_shape = in RewriteDynamicSort() local
H A Dselect_and_scatter_expander.cc35 auto operand_shape = operand->shape(); in ExpandInstruction() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Ddynamic_ops_test.cc511 void RunR3Contiguous(std::vector<int32_t> operand_shape, int32_t index, in RunR3Contiguous()
671 std::vector<int32_t> operand_shape({4, 5, 2}); in XLA_TEST_F() local
677 std::vector<int32_t> operand_shape({4, 5, 2}); in XLA_TEST_F() local
683 std::vector<int32_t> operand_shape({4, 5, 2}); in XLA_TEST_F() local
689 std::vector<int32_t> operand_shape({4, 5, 2}); in XLA_TEST_F() local
695 std::vector<int32_t> operand_shape({4, 5, 2}); in XLA_TEST_F() local
701 std::vector<int32_t> operand_shape({4, 5, 2}); in XLA_TEST_F() local
707 std::vector<int32_t> operand_shape({4, 5, 2}); in XLA_TEST_F() local
713 std::vector<int32_t> operand_shape({4, 5, 2}); in XLA_TEST_F() local
718 std::vector<int32_t> operand_shape({3, 123, 247}); in XLA_TEST_F() local
[all …]
H A Dselect_and_scatter_test.cc44 std::vector<int64_t> operand_shape; member
75 auto operand_shape = GetParam().operand_shape; in XLA_TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/shape_inference/
H A Dpass.cc109 ShapedType operand_shape = op->getOperand(0).getType().cast<ShapedType>(); in TryToCacheResultsTensorValue() local
122 ShapedType operand_shape = op->getOperand(0).getType().cast<ShapedType>(); in TryToCacheResultsTensorValue() local
141 auto operand_shape = operand_type.cast<ShapedType>(); in TryToCacheResultsTensorValue() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
H A Dir_array.cc297 const Shape& operand_shape, absl::Span<const int64_t> starts, in SourceIndexOfSlice()
315 const Shape& shape, const Shape& operand_shape, in SourceIndexOfTranspose()
330 const Shape& shape, const Shape& operand_shape, in SourceIndexOfBitcast()
367 const Shape& shape, const Shape& operand_shape, in SourceIndexOfBroadcast()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emission_utils.cc223 const Shape& operand_shape, absl::Span<int64_t const> dims_to_reduce) { in IsReductionFromOrToContiguousDimensionsImpl()
257 Shape operand_shape = GetShape(first_operand); in IsReductionFromOrToContiguousDimensions() local
302 Shape operand_shape = GetShape(input); in GetReductionKindAndContiguousComponents() local
878 const Shape& operand_shape = in Match021Transpose() local
H A Dreduction_layout_normalizer.cc69 const Shape &operand_shape = operand->shape(); in HandleReduce() local
H A Dir_emitter_unnested.h482 Shape operand_shape; member
H A Dir_emitter_unnested.cc452 const Shape& operand_shape = slice->operands()[0]->shape(); in GetConsistentInputShapeForRootSlices() local
1375 const Shape operand_shape = GetShape(fft_op.getOperand()); in EmitFftThunk() local
1835 auto operand_shape = GetShape(operands[0]); in EmitFusion() local
1904 const Shape operand_shape = GetShape(select_and_scatter_op.getOperand()); in EmitSelectAndScatter() local
2284 const Shape operand_shape = GetShape(scatter.getOperand()); in EmitScatter() local
3746 const Shape& operand_shape = reduction->inputs()[output_idx]->shape(); in GetOutputAddressForReduction() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dfuse_binary_into_preceding_affine.cc58 const Shape& operand_shape = operand.shape(); in FuseAddOrSubParamsIntoPrecedingAffine() local
128 const Shape& operand_shape = operand.shape(); in FuseMulOrDivParamsIntoPrecedingAffine() local
H A Dfuse_binary_into_following_affine.cc202 const auto& operand_shape = in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Ddynamic_shaped_ops.cc140 const XlaComputation& computation) -> StatusOr<XlaComputation> { in DynamicConditional()
200 const XlaComputation& computation) -> StatusOr<XlaComputation> { in DynamicConditional()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Daggregate_ops.cc64 xla::Shape operand_shape; in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_layout_assignment.cc144 Shape operand_shape( in AddBackendConstraints() local
H A Dir_emitter.cc537 const Shape& operand_shape = operand->shape(); in HandleOutfeed() local
1160 const Shape& operand_shape = crs->operand(i)->shape(); in HandleAllReduceSingleReplica() local
1223 const Shape& operand_shape = crs->operand(i)->shape(); in HandleAllReduceMultipleReplica() local
1307 const Shape& operand_shape = instruction->operand(i)->shape(); in HandleAllToAll() local
1428 const Shape& operand_shape = reduce.operand(0)->shape(); in ReductionPreservesLayout() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
H A Dshape_simplification.cc120 operand_extents, [&](ArrayRef<int64_t> operand_shape) { in matchAndRewrite()

12