Home
last modified time | relevance | path

Searched defs:operand_shape (Results 1 – 25 of 49) 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.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 …]
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
Dreduce_decomposer.cc141 auto operand_shape = operand->shape(); in ExpectedOutputShape() local
Dselect_and_scatter_expander.cc35 auto operand_shape = operand->shape(); in ExpandInstruction() local
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
Dbatchnorm_expander.cc172 const Shape operand_shape = operand->shape(); in HandleBatchNormTraining() local
301 const Shape operand_shape = operand->shape(); in HandleBatchNormInference() local
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()
Dindexed_array_analysis.cc351 absl::Span<const int64_t> operand_shape, in ComputeReshapePassthroughDimPairs()
450 absl::Span<const int64_t> operand_shape, in FindSourcePositionForPassthroughResultDim()
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
/external/tensorflow/tensorflow/compiler/xla/tests/
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 …]
Dselect_and_scatter_test.cc44 std::vector<int64_t> operand_shape; member
75 auto operand_shape = GetParam().operand_shape; in XLA_TEST_P() local
/external/tensorflow/tensorflow/core/transforms/shape_inference/
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
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
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()
/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.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/client/lib/
Ddynamic_shaped_ops.cc140 const XlaComputation& computation) -> StatusOr<XlaComputation> { in DynamicConditional()
200 const XlaComputation& computation) -> StatusOr<XlaComputation> { in DynamicConditional()
Dpooling_test.cc39 Shape operand_shape = b->GetShape(input).ValueOrDie(); in MakeGeneralPadding() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
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
Dreduction_layout_normalizer.cc69 const Shape &operand_shape = operand->shape(); in HandleReduce() local
Dir_emitter_unnested.h482 Shape operand_shape; member
Delemental_ir_emitter.cc164 Shape operand_shape = hlo->operand(0)->shape(); in GetSourceIndexOfBitcast() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment.cc144 Shape operand_shape( in AddBackendConstraints() local
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dshape_simplification.cc120 operand_extents, [&](ArrayRef<int64_t> operand_shape) { in matchAndRewrite()

12