| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_verifier.cc | 175 std::vector<const Shape*> operand_shapes; in HandleConcatenate() local
384 std::vector<const Shape*> operand_shapes; in HandleAllGather() local
397 std::vector<const Shape*> operand_shapes; in HandleAllGatherStart() local
419 std::vector<const Shape*> operand_shapes; in HandleAllReduce() local
456 std::vector<const Shape*> operand_shapes; in HandleReduceScatter() local
473 std::vector<const Shape*> operand_shapes; in HandleAllReduceStart() local
504 std::vector<const Shape*> operand_shapes; in HandleAllToAll() local
710 std::vector<const Shape*> operand_shapes; in HandleCollectivePermute() local
725 std::vector<const Shape*> operand_shapes; in HandleCollectivePermuteStart() local
1029 std::vector<const Shape*> operand_shapes; in HandleReduce() local
[all …]
|
| H A D | shape_inference.cc | 1069 std::vector<const Shape*> operand_shapes; in InferVariadicOpShape() local
1078 HloOpcode opcode, absl::Span<const Shape* const> operand_shapes) { in InferVariadicOpShape()
2036 absl::Span<const Shape* const> operand_shapes, int64_t all_gather_dimension, in InferAllGatherShape()
2060 absl::Span<const Shape* const> operand_shapes, int64_t all_gather_dimension, in InferAllGatherStartShape()
2080 absl::Span<const Shape* const> operand_shapes) { in InferAllReduceShape()
2092 absl::Span<const Shape* const> operand_shapes, int64_t scatter_dimension, in InferReduceScatterShape()
2127 absl::Span<const Shape* const> operand_shapes) { in InferAllReduceStartShape()
2165 absl::Span<const Shape* const> operand_shapes) { in InferAllToAllTupleShape()
2184 absl::Span<const Shape* const> operand_shapes) { in InferCollectivePermuteShape()
2196 absl::Span<const Shape* const> operand_shapes) { in InferCollectivePermuteStartShape()
[all …]
|
| H A D | async_collective_creator.cc | 84 std::vector<const Shape*> operand_shapes; in Run() local
126 std::vector<const Shape*> operand_shapes; in Run() local
|
| H A D | all_reduce_combiner.cc | 63 std::vector<const Shape*> operand_shapes; in CombineAllReduces() local
|
| H A D | hlo_creation_utils.cc | 282 std::vector<const Shape*> operand_shapes; in MakeConcatHlo() local
357 std::vector<const Shape*> operand_shapes; in MakeMapHlo() local
|
| H A D | hlo_evaluator.cc | 3715 absl::InlinedVector<const Shape*, 1> operand_shapes; in HandleReduce() local
|
| H A D | hlo_instruction.cc | 723 std::vector<Shape> operand_shapes; in CreateFromProto() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | py_client.cc | 481 absl::Span<Shape const> operand_shapes) { in CreateCallbackArgs()
542 pybind11::function callable, absl::Span<Shape const> operand_shapes, in MakePythonCallbackUsingHostSendAndRecv()
591 pybind11::function callable, absl::Span<Shape const> operand_shapes, in GetEmitPythonCallbackDescriptor()
692 std::vector<Shape> operand_shapes(operands.size()); in EmitPythonCallback() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gpu_autotuning.proto | 13 repeated xla.ShapeProto operand_shapes = 2; field
|
| H A D | ir_emission_utils.cc | 784 const std::vector<Shape>& operand_shapes, const Shape& output_shape) { in FindTranspose021DimsAndParameters()
|
| H A D | ir_emitter_unnested.cc | 1504 std::vector<Shape>* operand_shapes, in ProcessFusionForConversion()
2448 std::vector<Shape> operand_shapes, output_shapes; in GetOrCreateSubComputationFromRegion() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | fold_broadcast.cc | 167 std::array<llvm::ArrayRef<int64_t>, 2> operand_shapes; in RewriteOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | mlir_emitter.cc | 95 llvm::ArrayRef<Shape> operand_shapes, llvm::Value *result_ptr, in EmitMlirFuncAndCall()
|
| H A D | dot_op_emitter.cc | 266 Shape operand_shapes[] = {dot_info_.lhs_shape, dot_info_.rhs_shape}; in EmitLinalgMatmul() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | xla_builder.cc | 233 std::vector<Shape> operand_shapes; in GetOperandShapes() local
2450 absl::InlinedVector<const Shape*, 3> operand_shapes; in Scatter() local
2711 std::vector<const Shape*> operand_shapes, init_shapes; in ReduceWindowWithGeneralPadding() local
2756 std::vector<const Shape*> operand_shapes, init_shapes; in ReduceWindowInternal() local
2944 std::vector<const Shape*> operand_shapes; in AllReduce() local
3026 std::vector<const Shape*> operand_shapes; in ReduceScatter() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | spmd_partitioner.cc | 2841 std::vector<const Shape*> operand_shapes; in HandleSingleDevice() local
|