Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc426 const Shape& operand_shape, PrimitiveType new_element_type) { in InferConvertShape()
450 const Shape& operand_shape, PrimitiveType new_element_type) { in InferBitcastConvertShape()
480 const Shape& operand_shape, const int exponent_bits, in InferReducePrecisionShape()
504 const Shape& operand_shape, const Shape& padding_value_shape, in InferPadShape()
1084 for (const Shape* operand_shape : operand_shapes) { in InferVariadicOpShape() local
1191 const Shape& operand_shape, const Shape& scale_shape, in InferBatchNormTrainingShape()
1296 const Shape& operand_shape, const Shape& scale_shape, in InferBatchNormInferenceShape()
1440 const Shape& operand_shape, const Shape& scale_shape, in InferBatchNormGradShape()
2003 for (const Shape* operand_shape : operand_shapes) { in InferAllReduceShape() local
2011 for (const Shape* operand_shape : operand_shapes) { in InferAllReduceShape() local
[all …]
Dhlo_verifier.cc542 const Shape& operand_shape = instruction.operands()[i]->shape(); in SameElementTypesForOperandsAndToApplyParameters() local
600 const Shape& operand_shape = broadcast->operand(0)->shape(); in HandleBroadcast() local
618 const Shape& operand_shape = reshape->operand(0)->shape(); in HandleReshape() local
857 const Shape& operand_shape = copy_done->operand(0)->shape(); in HandleCopyDone() local
1578 const Shape& operand_shape = operand->shape(); in CheckElementwiseInstruction() local
1712 const Shape& operand_shape = operand->shape(); in Postprocess() local
Dhlo_creation_utils.cc383 const Shape& operand_shape = operand->shape(); in CollapseFirstNDims() local
408 const Shape& operand_shape = operand->shape(); in PrependDegenerateDims() local
445 const Shape& operand_shape = operand->shape(); in InsertDegenerateDims() local
Ddynamic_padder.cc250 const Shape operand_shape = reshape->operand(0)->shape(); in RewriteDynamicReshapeSplitInput() local
498 const Shape operand_shape = reshape->operand(0)->shape(); in RewriteDynamicReshapeSingleDim() local
648 Shape operand_shape = in RewriteDynamicSort() local
Dbatchnorm_expander.cc173 const Shape operand_shape = operand->shape(); in HandleBatchNormTraining() local
300 const Shape operand_shape = operand->shape(); in HandleBatchNormInference() local
Dindexed_array_analysis.cc345 absl::Span<const int64> operand_shape, in ComputeReshapePassthroughDimPairs()
444 absl::Span<const int64> operand_shape, absl::Span<const int64> result_shape, in FindSourcePositionForPassthroughResultDim()
Dhlo_cost_analysis_test.cc865 Shape operand_shape = ShapeUtil::MakeShape(S32, {3, 3}); in TEST_F() local
895 Shape operand_shape = ShapeUtil::MakeShape(F32, {3, 3}); in TEST_F() local
Dhlo_evaluator.cc490 const Shape& operand_shape = operands[i]->shape(); in HandleConcatenate() local
504 const Shape& operand_shape = operand->shape(); in HandleConcatenate() local
1741 const Shape& operand_shape = operand.shape(); in HandleGather() local
Dmemory_space_assignment.cc1328 Shape operand_shape = use.instruction->operand(use.operand_number)->shape(); in Process() local
Dlayout_assignment.cc1079 Shape operand_shape = operand->shape(); in ChooseOperandLayoutFromOutputLayout() local
Delemental_ir_emitter.cc1807 const Shape& operand_shape = hlo->operand(0)->shape(); in EmitElementalGather() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Ddynamic_ops_test.cc511 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 …]
Dselect_and_scatter_test.cc42 std::vector<int64> operand_shape; member
73 auto operand_shape = GetParam().operand_shape; in XLA_TEST_P() local
Dgather_operation_test.cc640 Shape operand_shape = ShapeUtil::MakeShape(S32, {3, 3}); in XLA_TEST_F() local
Dtest_utils.cc580 const Shape& operand_shape = use->operand(0)->shape(); in CreateLiteralForConstrainedUses() local
Dreduce_test.cc940 Shape operand_shape = ShapeUtil::MakeShape(F32, {1}); in XLA_TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_array.cc186 const Shape& operand_shape, absl::Span<const int64> starts, in SourceIndexOfSlice()
207 const Shape& shape, const Shape& operand_shape, in SourceIndexOfTranspose()
223 const Shape& shape, const Shape& operand_shape, in SourceIndexOfBitcast()
252 const Shape& shape, const Shape& operand_shape, in SourceIndexOfBroadcast()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Daggregate_ops.cc64 xla::Shape operand_shape; in Compile() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dfuse_binary_into_preceding_affine.cc45 const Shape& operand_shape = operand.shape(); in FuseAddOrSubParamsIntoPrecedingAffine() local
114 const Shape& operand_shape = operand.shape(); in FuseMulOrDivParamsIntoPrecedingAffine() local
Dfuse_binary_into_following_affine.cc202 const auto& operand_shape = in Run() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dreduction_layout_normalizer.cc41 const Shape &operand_shape = operand->shape(); in HandleReduce() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment.cc143 Shape operand_shape( in AddBackendConstraints() local
Dir_emitter.cc541 const Shape& operand_shape = operand->shape(); in HandleOutfeed() local
1394 const Shape& operand_shape = crs->operand(i)->shape(); in HandleAllReduceSingleReplica() local
1558 const Shape& operand_shape = reduce.operand(0)->shape(); in ReductionPreservesLayout() local
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling_test.cc39 Shape operand_shape = b->GetShape(input).ValueOrDie(); in MakeGeneralPadding() local
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc1468 ArrayRef<int64_t> operand_shape; in BuildSelectV2Op() local

12