/external/tensorflow/tensorflow/compiler/xla/service/ |
D | shape_inference.cc | 407 const Shape& operand_shape, PrimitiveType new_element_type) { in InferConvertShape() 431 const Shape& operand_shape, PrimitiveType new_element_type) { in InferBitcastConvertShape() 461 const Shape& operand_shape, const int exponent_bits, in InferReducePrecisionShape() 485 const Shape& operand_shape, const Shape& padding_value_shape, in InferPadShape() 1060 for (const Shape* operand_shape : operand_shapes) { in InferVariadicOpShape() local 1167 const Shape& operand_shape, const Shape& scale_shape, in InferBatchNormTrainingShape() 1272 const Shape& operand_shape, const Shape& scale_shape, in InferBatchNormInferenceShape() 1416 const Shape& operand_shape, const Shape& scale_shape, in InferBatchNormGradShape() 1977 for (const Shape* operand_shape : operand_shapes) { in InferAllReduceShape() local 1985 for (const Shape* operand_shape : operand_shapes) { in InferAllReduceShape() local [all …]
|
D | hlo_creation_utils.cc | 314 const Shape& operand_shape = operand->shape(); in CollapseFirstNDims() local 339 const Shape& operand_shape = operand->shape(); in PrependDegenerateDims() local 392 const Shape& operand_shape = operand->shape(); in InsertDegenerateDims() local
|
D | hlo_verifier.cc | 429 const Shape& operand_shape = instruction.operands()[i]->shape(); in SameElementTypesForOperandsAndToApplyParameters() local 478 const Shape& operand_shape = broadcast->operand(0)->shape(); in HandleBroadcast() local 496 const Shape& operand_shape = reshape->operand(0)->shape(); in HandleReshape() local 1310 const Shape& operand_shape = operand->shape(); in CheckElementwiseInstruction() local 1444 const Shape& operand_shape = operand->shape(); in Postprocess() local
|
D | batchnorm_expander.cc | 206 const Shape operand_shape = operand->shape(); in HandleBatchNormTraining() local 333 const Shape operand_shape = operand->shape(); in HandleBatchNormInference() local
|
D | hlo_cost_analysis_test.cc | 645 Shape operand_shape = ShapeUtil::MakeShape(S32, {3, 3}); in TEST_F() local 670 Shape operand_shape = ShapeUtil::MakeShape(F32, {3, 3}); in TEST_F() local
|
D | indexed_array_analysis.cc | 345 absl::Span<const int64> operand_shape, in ComputeReshapePassthroughDimPairs() 444 absl::Span<const int64> operand_shape, absl::Span<const int64> result_shape, in FindSourcePositionForPassthroughResultDim()
|
D | hlo_evaluator.cc | 465 const Shape& operand_shape = operands[i]->shape(); in HandleConcatenate() local 479 const Shape& operand_shape = operand->shape(); in HandleConcatenate() local 1083 const Shape& operand_shape = operand.shape(); in HandleGather() local
|
D | layout_assignment.cc | 1022 Shape operand_shape = operand->shape(); in ChooseOperandLayoutFromOutputLayout() local
|
D | hlo_evaluator_typed_visitor.h | 2321 const Shape& operand_shape = operand.shape(); in HandleScatter() local
|
D | elemental_ir_emitter.cc | 1863 const Shape& operand_shape = hlo->operand(0)->shape(); in EmitElementalGather() local
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | dynamic_ops_test.cc | 511 void RunR3Contiguous(std::vector<int32> operand_shape, int32 index, in RunR3Contiguous() 667 std::vector<int32> operand_shape({4, 5, 2}); in XLA_TEST_F() local 673 std::vector<int32> operand_shape({4, 5, 2}); in XLA_TEST_F() local 679 std::vector<int32> operand_shape({4, 5, 2}); in XLA_TEST_F() local 685 std::vector<int32> operand_shape({4, 5, 2}); in XLA_TEST_F() local 691 std::vector<int32> operand_shape({4, 5, 2}); in XLA_TEST_F() local 697 std::vector<int32> operand_shape({4, 5, 2}); in XLA_TEST_F() local 703 std::vector<int32> operand_shape({4, 5, 2}); in XLA_TEST_F() local 709 std::vector<int32> operand_shape({4, 5, 2}); in XLA_TEST_F() local 714 std::vector<int32> operand_shape({3, 123, 247}); in XLA_TEST_F() local [all …]
|
D | select_and_scatter_test.cc | 42 std::vector<int64> operand_shape; member 73 auto operand_shape = GetParam().operand_shape; in XLA_TEST_P() local
|
D | test_utils.cc | 469 const Shape& operand_shape = use->operand(0)->shape(); in CreateLiteralForConstrainedUses() local
|
D | gather_operation_test.cc | 641 Shape operand_shape = ShapeUtil::MakeShape(S32, {3, 3}); in XLA_TEST_F() local
|
D | reduce_test.cc | 939 Shape operand_shape = ShapeUtil::MakeShape(F32, {1}); in XLA_TEST_F() local
|
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
D | ir_array.cc | 181 const Shape& operand_shape, absl::Span<const int64> starts, in SourceIndexOfSlice() 202 const Shape& shape, const Shape& operand_shape, in SourceIndexOfTranspose() 218 const Shape& shape, const Shape& operand_shape, in SourceIndexOfBitcast() 251 const Shape& shape, const Shape& operand_shape, in SourceIndexOfBroadcast()
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | aggregate_ops.cc | 47 TensorShape operand_shape; in Compile() local
|
/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
D | fuse_binary_into_preceding_affine.cc | 45 const Shape& operand_shape = operand.shape(); in FuseAddOrSubParamsIntoPrecedingAffine() local 114 const Shape& operand_shape = operand.shape(); in FuseMulOrDivParamsIntoPrecedingAffine() local
|
D | fuse_binary_into_following_affine.cc | 202 const auto& operand_shape = in Run() local
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | cpu_layout_assignment.cc | 143 Shape operand_shape( in AddBackendConstraints() local
|
D | ir_emitter.cc | 482 const Shape& operand_shape = operand->shape(); in HandleOutfeed() local 1337 const Shape& operand_shape = crs->operand(i)->shape(); in HandleAllReduce() local 1370 const Shape& operand_shape = reduce.operand(0)->shape(); in ReductionPreservesLayout() local
|
/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
D | pooling_test.cc | 39 Shape operand_shape = b->GetShape(input).ValueOrDie(); in MakeGeneralPadding() local
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder.cc | 218 Shape operand_shape(operand.shape()); in IsConstantVisitor() local
|