Searched defs:operand_shapes (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_autotuning.proto | 13 repeated xla.ShapeProto operand_shapes = 2; field
|
| D | ir_emitter_unnested.h | 81 std::vector<Shape> operand_shapes; member
|
| D | ir_emitter_unnested.cc | 1684 std::vector<Shape>* operand_shapes, in ProcessFusionForConversion()
2739 std::vector<Shape> operand_shapes, output_shapes; in GetOrCreateSubComputationFromRegion() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_verifier.cc | 115 std::vector<const Shape*> operand_shapes; in HandleConcatenate() local
284 std::vector<const Shape*> operand_shapes; in HandleAllReduce() local
326 std::vector<const Shape*> operand_shapes; in HandleAllToAll() local
666 std::vector<const Shape*> operand_shapes; in HandleReduce() local
870 std::vector<const Shape*> operand_shapes; in HandleMap() local
1149 std::vector<const Shape*> operand_shapes; in HandleAfterAll() local
|
| D | all_reduce_combiner.cc | 62 std::vector<Shape> operand_shapes; in CombineAllReduces() local
|
| 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 | shape_inference.cc | 1067 std::vector<const Shape*> operand_shapes; in InferVariadicOpShape() local
1076 HloOpcode opcode, absl::Span<const Shape* const> operand_shapes) { in InferVariadicOpShape()
2046 absl::Span<const Shape* const> operand_shapes) { in InferAllReduceShape()
2089 absl::Span<const Shape* const> operand_shapes) { in InferAllToAllTupleShape()
|
| D | hlo_evaluator.cc | 2410 absl::InlinedVector<const Shape*, 1> operand_shapes; in HandleReduce() local
|
| D | hlo_instruction.cc | 554 std::vector<Shape> operand_shapes; in CreateFromProto() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | fold_broadcast.cc | 165 std::array<llvm::ArrayRef<int64_t>, 2> operand_shapes; in RewriteOp() 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 | 255 Shape operand_shapes[] = {dot_info_.lhs_shape, dot_info_.rhs_shape}; in EmitLinalgMatmul() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 235 std::vector<Shape> operand_shapes; in GetOperandShapes() local
2557 std::vector<const Shape*> operand_shapes, init_shapes; in ReduceWindowWithGeneralPadding() local
2602 std::vector<const Shape*> operand_shapes, init_shapes; in ReduceWindowInternal() local
2788 std::vector<const Shape*> operand_shapes; in AllReduce() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 2184 std::vector<Shape> operand_shapes; in HandleSingleDevice() local
|