| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 67 auto inferred_status = ShapeInference::InferReduceShape( in ExpectInferredReduceShape()
108 ShapeInference::InferUnaryOpShape(HloOpcode::kNegate, matrix_shape); in TEST_F()
115 auto inferred_status = ShapeInference::InferTernaryOpShape( in TEST_F()
123 auto inferred_status = ShapeInference::InferTernaryOpShape( in TEST_F()
133 auto inferred_status = ShapeInference::InferTernaryOpShape( in TEST_F()
140 auto inferred_status_error1 = ShapeInference::InferTernaryOpShape( in TEST_F()
146 auto inferred_status_error2 = ShapeInference::InferTernaryOpShape( in TEST_F()
152 auto inferred_status_error3 = ShapeInference::InferTernaryOpShape( in TEST_F()
161 auto inferred_status_error4 = ShapeInference::InferTernaryOpShape( in TEST_F()
171 auto inferred_status = ShapeInference::InferTernaryOpShape( in TEST_F()
[all …]
|
| D | shape_inference.cc | 249 /* static */ StatusOr<Shape> ShapeInference::InferUnaryOpShape( in InferUnaryOpShape()
254 /* static */ StatusOr<Shape> ShapeInference::InferUnaryOpShape( in InferUnaryOpShape()
382 /* static */ StatusOr<Shape> ShapeInference::InferConcatOpShape( in InferConcatOpShape()
451 /* static */ StatusOr<Shape> ShapeInference::InferConvertShape( in InferConvertShape()
475 /* static */ StatusOr<Shape> ShapeInference::InferBitcastConvertShape( in InferBitcastConvertShape()
505 /* static */ StatusOr<Shape> ShapeInference::InferReducePrecisionShape( in InferReducePrecisionShape()
529 /* static */ StatusOr<Shape> ShapeInference::InferPadShape( in InferPadShape()
648 /* static */ StatusOr<Shape> ShapeInference::InferDotOpShape( in InferDotOpShape()
741 ShapeInference::InferDegenerateDimensionBroadcastShape(HloOpcode operation, in InferDegenerateDimensionBroadcastShape()
778 /* static */ StatusOr<Shape> ShapeInference::InferInDimBroadcastShape( in InferInDimBroadcastShape()
[all …]
|
| D | hlo_verifier.cc | 120 ShapeInference::InferConcatOpShape( in HandleConcatenate()
125 return CheckShape(convert, ShapeInference::InferConvertShape( in HandleConvert()
131 return CheckShape(convert, ShapeInference::InferBitcastConvertShape( in HandleBitcastConvert()
143 ShapeInference::InferDotOpShape( in HandleDot()
153 ShapeInference::InferConvolveShape( in HandleConvolution()
164 ShapeInference::InferFftShape(fft->operand(0)->shape(), fft->fft_type(), in HandleFft()
171 ShapeInference::InferTriangularSolveShape( in HandleTriangularSolve()
179 TF_ASSIGN_OR_RETURN(const Shape expected, ShapeInference::InferCholeskyShape( in HandleCholesky()
275 return CheckShape(ag, ShapeInference::InferAllGatherShape( in HandleAllGather()
288 return CheckShape(hlo, ShapeInference::InferAllReduceShape(operand_shapes)); in HandleAllReduce()
[all …]
|
| D | hlo_creation_utils.cc | 41 ShapeInference::InferUnaryOpShape(opcode, operand)); in MakeUnaryHlo()
51 ShapeInference::InferBinaryOpShape(opcode, lhs, rhs)); in MakeBinaryHlo()
63 ShapeInference::InferBinaryOpShape(HloOpcode::kCompare, lhs, rhs)); in MakeCompareHlo()
75 ShapeInference::InferPadShape(operand->shape(), padding_value->shape(), in MakePadHlo()
86 TF_ASSIGN_OR_RETURN(Shape slice_shape, ShapeInference::InferSliceShape( in MakeSliceHlo()
103 ShapeInference::InferConvolveShape( in MakeConvolveHlo()
116 ShapeInference::InferTransposeShape(operand->shape(), dimensions)); in MakeTransposeHlo()
144 ShapeInference::InferDynamicSliceShape( in MakeDynamicSliceHlo()
171 ShapeInference::InferDynamicSliceShape( in MakeDynamicSliceHlo()
199 ShapeInference::InferDynamicUpdateSliceShape( in MakeDynamicUpdateSliceHlo()
[all …]
|
| D | operand_upcaster.cc | 27 return ShapeInference::InferDotOpShape( in MaybeInferShape()
32 return ShapeInference::InferConvolveShape( in MaybeInferShape()
|
| D | tree_reduction_rewriter.cc | 76 Window window, ShapeInference::InferWindowFromDimensions( in HandleReduce()
80 ShapeInference::InferReduceWindowShape( in HandleReduce()
|
| D | hlo_parser.cc | 1115 return ShapeInference::InferUnaryOpShape(opcode, operands[0]); in ParseInstructionRhs()
1147 return ShapeInference::InferBinaryOpShape(opcode, operands[0], in ParseInstructionRhs()
1167 return ShapeInference::InferTernaryOpShape( in ParseInstructionRhs()
1413 return ShapeInference::InferVariadicOpShape(opcode, arg_shapes); in ParseInstructionRhs()
1443 return ShapeInference::InferWhileShape( in ParseInstructionRhs()
1560 return ShapeInference::InferCallShape( in ParseInstructionRhs()
1589 return ShapeInference::InferReduceWindowShape( in ParseInstructionRhs()
1644 return ShapeInference::InferConvolveShape( in ParseInstructionRhs()
1670 return ShapeInference::InferFftShape(operands[0]->shape(), in ParseInstructionRhs()
1689 return ShapeInference::InferTriangularSolveShape( in ParseInstructionRhs()
[all …]
|
| D | transpose_folding_test.cc | 243 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape( in TEST_F()
300 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape( in TEST_F()
362 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape( in TEST_F()
430 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape( in TEST_F()
|
| D | shape_inference.h | 45 class ShapeInference {
382 TF_DISALLOW_COPY_AND_ASSIGN(ShapeInference);
|
| D | dynamic_window_utils.cc | 54 ShapeInference::InferBinaryOpShape(opcode, x.get(), y.get()).ValueOrDie(); in BinaryOp()
|
| D | dot_as_convolution_util.cc | 142 ShapeInference::InferConvolveShape( in CreateShardedConvForDotGeneralConvolution()
|
| D | hlo_evaluator_typed_visitor.h | 1062 ShapeInference::InferReverseShape(operand->shape(),
1291 ShapeInference::InferConvolveShape(
1541 ShapeInference::InferPadShape(
1606 ShapeInference::InferDynamicSliceShape(
1669 ShapeInference::InferDynamicUpdateSliceShape(
1951 ShapeInference::InferReduceWindowShape(
2439 ShapeInference::InferSliceShape(
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | shape_inference_pass.cc | 31 class ShapeInference : public TensorFlowShapeInferencePassBase<ShapeInference> { class
42 return std::make_unique<ShapeInference>(); in CreateTFShapeInferencePass()
|
| D | shape_inference.cc | 411 class ShapeInference { class
413 ShapeInference(int64_t graph_version, MLIRContext* context,
617 ShapeInference::ShapeInference(int64_t graph_version, MLIRContext* context, in ShapeInference() function in mlir::TF::ShapeInference
623 ArrayRef<Operation*> ShapeInference::GetCallers(FuncOp fn) { in GetCallers()
638 void ShapeInference::EnqueueCallers(FuncOp fn) { in EnqueueCallers()
642 void ShapeInference::UpdateTypeAndInsertIncompatibleUseCasts(Type new_type, in UpdateTypeAndInsertIncompatibleUseCasts()
671 bool ShapeInference::RefineResultType(Operation* op, Value result, in RefineResultType()
682 bool ShapeInference::InferShapeForCall(CallOpInterface call_op) { in InferShapeForCall()
700 bool ShapeInference::InferShapeForCast(CastOp op) { in InferShapeForCast()
728 bool ShapeInference::InferShapeForIf(IfOp op) { in InferShapeForIf()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter_test.cc | 110 ShapeInference::InferConvolveShape( in TEST_F()
151 ShapeInference::InferConvolveShape( in TEST_F()
298 ShapeInference::InferConvolveShape( in TEST_F()
343 ShapeInference::InferConvolveShape( in TEST_F()
379 ShapeInference::InferConvolveShape( in TEST_F()
437 conv->shape(), ShapeInference::InferConvolveShape( in TEST_F()
488 conv->shape(), ShapeInference::InferConvolveShape( in TEST_F()
543 conv->shape(), ShapeInference::InferConvolveShape( in TEST_F()
599 conv->shape(), ShapeInference::InferConvolveShape( in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 620 Shape shape, ShapeInference::InferUnaryOpShape(unop, *operand_shape)); in UnaryOp()
633 Shape shape, ShapeInference::InferBinaryOpShape( in BinaryOp()
776 StatusOr<const Shape> status_or_shape = ShapeInference::InferTernaryOpShape( in TernaryOp()
833 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferCallShape( in Call()
872 ShapeInference::InferBroadcastShape(*operand_shape, broadcast_sizes)); in Broadcast()
917 TF_RETURN_IF_ERROR(ShapeInference::InferBroadcastShape( in BroadcastInDim()
957 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferSliceShape( in Slice()
1006 ShapeInference::InferDynamicSliceShape( in DynamicSlice()
1039 Shape shape, ShapeInference::InferDynamicUpdateSliceShape( in DynamicUpdateSlice()
1064 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferConcatOpShape( in ConcatInDim()
[all …]
|
| /external/llvm-project/mlir/examples/toy/Ch5/include/toy/ |
| D | ShapeInferenceInterface.td | 18 def ShapeInferenceOpInterface : OpInterface<"ShapeInference"> {
|
| /external/llvm-project/mlir/examples/toy/Ch4/mlir/ |
| D | ShapeInferencePass.cpp | 75 if (auto shapeOp = dyn_cast<ShapeInference>(op)) { in runOnFunction()
|
| /external/llvm-project/mlir/examples/toy/Ch6/include/toy/ |
| D | ShapeInferenceInterface.td | 18 def ShapeInferenceOpInterface : OpInterface<"ShapeInference"> {
|
| /external/llvm-project/mlir/examples/toy/Ch5/mlir/ |
| D | ShapeInferencePass.cpp | 75 if (auto shapeOp = dyn_cast<ShapeInference>(op)) { in runOnFunction()
|
| /external/llvm-project/mlir/examples/toy/Ch4/include/toy/ |
| D | ShapeInferenceInterface.td | 18 def ShapeInferenceOpInterface : OpInterface<"ShapeInference"> {
|
| /external/llvm-project/mlir/examples/toy/Ch7/include/toy/ |
| D | ShapeInferenceInterface.td | 18 def ShapeInferenceOpInterface : OpInterface<"ShapeInference"> {
|
| /external/llvm-project/mlir/examples/toy/Ch7/mlir/ |
| D | ShapeInferencePass.cpp | 75 if (auto shapeOp = dyn_cast<ShapeInference>(op)) { in runOnFunction()
|
| /external/llvm-project/mlir/examples/toy/Ch6/mlir/ |
| D | ShapeInferencePass.cpp | 75 if (auto shapeOp = dyn_cast<ShapeInference>(op)) { in runOnFunction()
|
| /external/llvm-project/mlir/docs/Tutorials/Toy/ |
| D | Ch-4.md | 268 purposes, we will simply name the generated class `ShapeInference`. We also
272 def ShapeInferenceOpInterface : OpInterface<"ShapeInference"> {
288 def ShapeInferenceOpInterface : OpInterface<"ShapeInference"> {
359 `ShapeInference` interface, using this code snippet:
366 if (ShapeInference shapeOp = dyn_cast<ShapeInference>(op)) {
|