Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc407 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 …]
Dhlo_creation_utils.cc314 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
Dhlo_verifier.cc429 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
Dbatchnorm_expander.cc206 const Shape operand_shape = operand->shape(); in HandleBatchNormTraining() local
333 const Shape operand_shape = operand->shape(); in HandleBatchNormInference() local
Dhlo_cost_analysis_test.cc645 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
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_evaluator.cc465 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
Dlayout_assignment.cc1022 Shape operand_shape = operand->shape(); in ChooseOperandLayoutFromOutputLayout() local
Dhlo_evaluator_typed_visitor.h2321 const Shape& operand_shape = operand.shape(); in HandleScatter() local
Delemental_ir_emitter.cc1863 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
Dtest_utils.cc469 const Shape& operand_shape = use->operand(0)->shape(); in CreateLiteralForConstrainedUses() local
Dgather_operation_test.cc641 Shape operand_shape = ShapeUtil::MakeShape(S32, {3, 3}); in XLA_TEST_F() local
Dreduce_test.cc939 Shape operand_shape = ShapeUtil::MakeShape(F32, {1}); in XLA_TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_array.cc181 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/
Daggregate_ops.cc47 TensorShape 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/cpu/
Dcpu_layout_assignment.cc143 Shape operand_shape( in AddBackendConstraints() local
Dir_emitter.cc482 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/
Dpooling_test.cc39 Shape operand_shape = b->GetShape(input).ValueOrDie(); in MakeGeneralPadding() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc218 Shape operand_shape(operand.shape()); in IsConstantVisitor() local