| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_verifier.cc | 151 std::vector<const Shape*> operand_shapes; in HandleConcatenate() local
355 std::vector<const Shape*> operand_shapes; in HandleAllGather() local
368 std::vector<const Shape*> operand_shapes; in HandleAllGatherStart() local
390 std::vector<const Shape*> operand_shapes; in HandleAllReduce() local
427 std::vector<const Shape*> operand_shapes; in HandleReduceScatter() local
444 std::vector<const Shape*> operand_shapes; in HandleAllReduceStart() local
488 std::vector<const Shape*> operand_shapes; in HandleAllToAll() local
694 std::vector<const Shape*> operand_shapes; in HandleCollectivePermute() local
709 std::vector<const Shape*> operand_shapes; in HandleCollectivePermuteStart() local
1003 std::vector<const Shape*> operand_shapes; in HandleReduce() local
[all …]
|
| D | async_collective_creator.cc | 58 std::vector<Shape> operand_shapes; in Run() local
|
| D | all_reduce_combiner.cc | 64 std::vector<Shape> operand_shapes; in CombineAllReduces() local
|
| D | shape_inference.cc | 1056 std::vector<const Shape*> operand_shapes; in InferVariadicOpShape() local
1065 HloOpcode opcode, absl::Span<const Shape* const> operand_shapes) { in InferVariadicOpShape()
2027 absl::Span<const Shape* const> operand_shapes, int64_t all_gather_dimension, in InferAllGatherShape()
2051 absl::Span<const Shape* const> operand_shapes, int64_t all_gather_dimension, in InferAllGatherStartShape()
2076 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()
2170 absl::Span<const Shape* const> operand_shapes) { in InferAllToAllTupleShape()
2189 absl::Span<const Shape* const> operand_shapes) { in InferCollectivePermuteShape()
2201 absl::Span<const Shape* const> operand_shapes) { in InferCollectivePermuteStartShape()
|
| D | hlo_creation_utils.cc | 242 std::vector<const Shape*> operand_shapes; in MakeConcatHlo() local
305 std::vector<const Shape*> operand_shapes; in MakeMapHlo() local
|
| D | hlo_evaluator.cc | 2474 absl::InlinedVector<const Shape*, 1> operand_shapes; in HandleReduce() local
|
| D | hlo_instruction.cc | 667 std::vector<Shape> operand_shapes; in CreateFromProto() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_autotuning.proto | 13 repeated xla.ShapeProto operand_shapes = 2; field
|
| D | ir_emitter_unnested.h | 54 std::vector<Shape> operand_shapes; member
|
| D | ir_emitter_unnested.cc | 1624 std::vector<Shape>* operand_shapes, in ProcessFusionForConversion()
2588 std::vector<Shape> operand_shapes, output_shapes; in GetOrCreateSubComputationFromRegion() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | mlir_emitter.cc | 92 llvm::ArrayRef<Shape> operand_shapes, llvm::Value *result_ptr, in EmitMlirFuncAndCall()
|
| D | dot_op_emitter.cc | 253 Shape operand_shapes[] = {dot_info_.lhs_shape, dot_info_.rhs_shape}; in EmitLinalgMatmul() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | fold_broadcast.cc | 172 std::array<llvm::ArrayRef<int64_t>, 2> operand_shapes; in RewriteOp() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 181 std::vector<Shape> operand_shapes; in GetOperandShapes() local
2585 std::vector<const Shape*> operand_shapes, init_shapes; in ReduceWindowWithGeneralPadding() local
2630 std::vector<const Shape*> operand_shapes, init_shapes; in ReduceWindowInternal() local
2817 std::vector<const Shape*> operand_shapes; in AllReduce() local
2895 std::vector<const Shape*> operand_shapes; in ReduceScatter() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 2146 std::vector<Shape> operand_shapes; in HandleSingleDevice() local
|