Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc441 const Shape& operand_shape, PrimitiveType new_element_type) { in InferConvertShape()
465 const Shape& operand_shape, PrimitiveType new_element_type) { in InferBitcastConvertShape()
495 const Shape& operand_shape, const int exponent_bits, in InferReducePrecisionShape()
519 const Shape& operand_shape, const Shape& padding_value_shape, in InferPadShape()
1093 for (const Shape* operand_shape : operand_shapes) { in InferVariadicOpShape() local
1200 const Shape& operand_shape, const Shape& scale_shape, in InferBatchNormTrainingShape()
1305 const Shape& operand_shape, const Shape& scale_shape, in InferBatchNormInferenceShape()
1449 const Shape& operand_shape, const Shape& scale_shape, in InferBatchNormGradShape()
2034 for (const Shape* operand_shape : operand_shapes) { in InferAllGatherShape() local
2077 for (const Shape* operand_shape : operand_shapes) { in InferAllReduceShape() local
[all …]
Dlogistic_expander.cc43 const Shape operand_shape = operand->shape(); in ExpandLogisticWithTanh() local
59 const Shape operand_shape = operand->shape(); in ExpandLogisticWithExp() local
Dhlo_verifier.cc984 const Shape& operand_shape = instruction.operands()[i]->shape(); in SameElementTypesForOperandsAndToApplyParameters() local
1036 const Shape& operand_shape = broadcast->operand(0)->shape(); in HandleBroadcast() local
1054 const Shape& operand_shape = dynamic_reshape->operand(0)->shape(); in HandleDynamicReshape() local
1068 const Shape& operand_shape = reshape->operand(0)->shape(); in HandleReshape() local
1329 const Shape& operand_shape = copy_done->operand(0)->shape(); in HandleCopyDone() local
2099 const Shape& operand_shape = operand->shape(); in CheckElementwiseInstruction() local
2111 const Shape& operand_shape = comparison->operand(1)->shape(); in CheckElementwiseInstruction() local
2258 const Shape& operand_shape = operand->shape(); in Postprocess() local
Dhlo_creation_utils.cc445 const Shape& operand_shape = operand->shape(); in CollapseFirstNDims() local
470 const Shape& operand_shape = operand->shape(); in PrependDegenerateDims() local
508 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.cc829 const Shape& operand_shape, const HloSharding& operand_sharding, in PassthroughOperandToGatherOutputOrScatterUpdate()
875 const Shape& operand_shape, const HloSharding& update_or_gather_sharding, in PassthroughGatherOutputOrScatterUpdateToOperand()
996 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.cc357 const Shape operand_shape = reshape->operand(0)->shape(); in RewriteDynamicReshapeSplitInput() local
724 const Shape operand_shape = reshape->operand(0)->shape(); in RewriteDynamicReshapeSingleGroup() local
1313 Shape operand_shape = in RewriteDynamicSort() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Ddynamic_ops_test.cc512 void RunR3Contiguous(std::vector<int32> operand_shape, int32_t 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/xla/service/llvm_ir/
Dir_array.cc294 const Shape& operand_shape, absl::Span<const int64> starts, in SourceIndexOfSlice()
312 const Shape& shape, const Shape& operand_shape, in SourceIndexOfTranspose()
327 const Shape& shape, const Shape& operand_shape, in SourceIndexOfBitcast()
364 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/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.cc112 auto operand_shape = operand.getType().template cast<MemRefType>().getShape(); in MapWindowIvsToInput() local
242 auto operand_shape = operand.getType().cast<MemRefType>().getShape(); in CreateReduceOpInNestedParallelLoops() local
Dhlo_legalize_to_memref.cc189 auto operand_shape = operand_type.getShape(); in InsertDynamicMemrefCastOp() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dreduction_layout_normalizer.cc41 const Shape &operand_shape = operand->shape(); in HandleReduce() local
Dir_emitter_unnested.h409 Shape operand_shape; member
Dir_emission_utils.cc396 const Shape operand_shape = layout_analysis.GetShape(input); in IsReductionFromOrToContiguousDimensions() local
491 Shape operand_shape = GetShape(input); in GetReductionKindAndContiguousComponents() local
Delemental_ir_emitter.cc166 Shape operand_shape = hlo->operand(0)->shape(); in GetSourceIndexOfBitcast() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment.cc143 Shape operand_shape( in AddBackendConstraints() local
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Ddynamic_shaped_ops.cc89 const XlaComputation& computation) { in DynamicConditional()
Dpooling_test.cc39 Shape operand_shape = b->GetShape(input).ValueOrDie(); in MakeGeneralPadding() local
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dshape_simplification.cc120 operand_extents, [&](ArrayRef<int64_t> operand_shape) { in matchAndRewrite()

12