Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dlogistic_expander.cc43 const Shape operand_shape = operand->shape(); in ExpandLogisticWithTanh() local
59 const Shape operand_shape = operand->shape(); in ExpandLogisticWithExp() local
Dshape_inference.cc452 const Shape& operand_shape, PrimitiveType new_element_type) { in InferConvertShape()
476 const Shape& operand_shape, PrimitiveType new_element_type) { in InferBitcastConvertShape()
506 const Shape& operand_shape, const int exponent_bits, in InferReducePrecisionShape()
530 const Shape& operand_shape, const Shape& padding_value_shape, in InferPadShape()
1104 for (const Shape* operand_shape : operand_shapes) { in InferVariadicOpShape() local
1211 const Shape& operand_shape, const Shape& scale_shape, in InferBatchNormTrainingShape()
1316 const Shape& operand_shape, const Shape& scale_shape, in InferBatchNormInferenceShape()
1460 const Shape& operand_shape, const Shape& scale_shape, in InferBatchNormGradShape()
2035 const Shape& operand_shape, int64 all_gather_dimension, int64 shard_count) { in InferAllGatherShape()
2047 for (const Shape* operand_shape : operand_shapes) { in InferAllReduceShape() local
[all …]
Dhlo_verifier.cc647 const Shape& operand_shape = instruction.operands()[i]->shape(); in SameElementTypesForOperandsAndToApplyParameters() local
699 const Shape& operand_shape = broadcast->operand(0)->shape(); in HandleBroadcast() local
717 const Shape& operand_shape = dynamic_reshape->operand(0)->shape(); in HandleDynamicReshape() local
731 const Shape& operand_shape = reshape->operand(0)->shape(); in HandleReshape() local
992 const Shape& operand_shape = copy_done->operand(0)->shape(); in HandleCopyDone() local
1747 const Shape& operand_shape = operand->shape(); in CheckElementwiseInstruction() local
1759 const Shape& operand_shape = comparison->operand(1)->shape(); in CheckElementwiseInstruction() local
1906 const Shape& operand_shape = operand->shape(); in Postprocess() local
Dhlo_creation_utils.cc442 const Shape& operand_shape = operand->shape(); in CollapseFirstNDims() local
467 const Shape& operand_shape = operand->shape(); in PrependDegenerateDims() local
504 const Shape& operand_shape = operand->shape(); in InsertDegenerateDims() local
Dbatchnorm_expander.cc173 const Shape operand_shape = operand->shape(); in HandleBatchNormTraining() local
300 const Shape operand_shape = operand->shape(); in HandleBatchNormInference() local
Dhlo_sharding_util.cc800 const Shape& operand_shape, const HloSharding& operand_sharding, in PassthroughOperandToGatherOutputOrScatterUpdate()
846 const Shape& operand_shape, const HloSharding& update_or_gather_sharding, in PassthroughGatherOutputOrScatterUpdateToOperand()
948 const Shape& output_shape, const Shape& operand_shape) { in GatherOutputShardingFromDataOperand()
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()
Ddynamic_padder.cc347 const Shape operand_shape = reshape->operand(0)->shape(); in RewriteDynamicReshapeSplitInput() local
714 const Shape operand_shape = reshape->operand(0)->shape(); in RewriteDynamicReshapeSingleGroup() local
1212 Shape operand_shape = in RewriteDynamicSort() local
Dhlo_cost_analysis_test.cc972 Shape operand_shape = ShapeUtil::MakeShape(S32, {3, 3}); in TEST_F() local
1002 Shape operand_shape = ShapeUtil::MakeShape(F32, {3, 3}); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Ddynamic_ops_test.cc512 void RunR3Contiguous(std::vector<int32> operand_shape, int32 index, in RunR3Contiguous()
668 std::vector<int32> operand_shape({4, 5, 2}); in XLA_TEST_F() local
674 std::vector<int32> operand_shape({4, 5, 2}); in XLA_TEST_F() local
680 std::vector<int32> operand_shape({4, 5, 2}); in XLA_TEST_F() local
686 std::vector<int32> operand_shape({4, 5, 2}); in XLA_TEST_F() local
692 std::vector<int32> operand_shape({4, 5, 2}); in XLA_TEST_F() local
698 std::vector<int32> operand_shape({4, 5, 2}); in XLA_TEST_F() local
704 std::vector<int32> operand_shape({4, 5, 2}); in XLA_TEST_F() local
710 std::vector<int32> operand_shape({4, 5, 2}); in XLA_TEST_F() local
715 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
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Daggregate_ops.cc64 xla::Shape operand_shape; in Compile() local
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_array.cc268 const Shape& operand_shape, absl::Span<const int64> starts, in SourceIndexOfSlice()
286 const Shape& shape, const Shape& operand_shape, in SourceIndexOfTranspose()
301 const Shape& shape, const Shape& operand_shape, in SourceIndexOfBitcast()
338 const Shape& shape, const Shape& operand_shape, in SourceIndexOfBroadcast()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dmaterialize_broadcasts.cc47 ArrayRef<int64_t> operand_shape = operand_type.getShape(); in matchAndRewrite() local
Dlhlo_legalize_to_parallel_loops.cc109 auto operand_shape = operand.getType().template cast<MemRefType>().getShape(); in MapWindowIvsToInput() local
239 auto operand_shape = operand.getType().cast<MemRefType>().getShape(); in CreateReduceOpInNestedParallelLoops() local
Dlegalize_to_linalg.cc462 auto operand_shape = operand_type.getShape(); in getIndexingMaps() local
593 auto operand_shape = operand_type.getShape(); in InsertReshapeIfNecessary() local
669 auto operand_shape = operand_type.getShape(); in getIndexingMaps() local
928 auto operand_shape = in matchAndRewrite() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
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
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
Dir_emitter_unnested.h479 Shape operand_shape; member
Dir_emission_utils.cc318 Shape operand_shape = TypeToShape(input.getType()); in IsReductionFromOrToContiguousDimensions() local
406 Shape operand_shape = TypeToShape(input.getType()); in GetReductionKindAndContiguousComponents() local
Dir_emitter_unnested.cc512 Shape operand_shape = TypeToShape(slice.operand().getType()); in GetConsistentInputShapeForRootSlices() local
1571 const Shape operand_shape = TypeToShape(fft_op.operand().getType()); in EmitFftThunkFromMlir() local
2089 auto operand_shape = TypeToShape(copy.operand().getType()); in EmitCopyForMlir() local
2234 const Shape operand_shape = in EmitSelectAndScatterFromMlir() local
2570 const Shape operand_shape = TypeToShape(scatter.operand().getType()); in EmitScatter() local
4425 auto operand_shape = TypeToShape(reduce_hlo->getOperand(0).getType()); in EmitEpilogueForReduction() local
5101 const Shape& operand_shape = context.operand_shapes[operand_idx]; in CheckAndEmitHloWithTile021() local
5157 const Shape& operand_shape = context.operand_shapes[params_012[i]]; in CheckAndEmitHloWithTile021() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment.cc143 Shape operand_shape( in AddBackendConstraints() local
Dir_emitter.cc523 const Shape& operand_shape = operand->shape(); in HandleOutfeed() local
1088 const Shape& operand_shape = crs->operand(i)->shape(); in HandleAllReduceSingleReplica() local
1149 const Shape& operand_shape = crs->operand(i)->shape(); in HandleAllReduceMultipleReplica() local
1228 const Shape& operand_shape = instruction->operand(i)->shape(); in HandleAllToAll() local
1335 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

12