| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
| H A D | shape_type_inference.cpp | 489 std::vector<c10::ShapeSymbol> final_shape; in ComputeShapeFromReshape() local
567 std::vector<::c10::ShapeSymbol> final_shape; in ComputeShapeFromExpand() local
609 std::vector<::c10::ShapeSymbol> final_shape; in ComputeShapeFromTile() local
710 c10::SymbolicShape final_shape(shape_size); in SetShapeValueFromListConstructNode() local
723 std::vector<::c10::ShapeSymbol> final_shape; in Broadcast() local
773 auto final_shape = Broadcast(input_shape_value_0, input_shape_value_1); in ProcessBroadcastNode() local
788 std::vector<::c10::ShapeSymbol> final_shape; in ProcessShapeForConcatNode() local
860 c10::SymbolicShape final_shape(shape_size); in ProcessShapeValueForConcatNode() local
900 auto final_shape = Broadcast( in ProcessMatMulNode() local
923 std::vector<::c10::ShapeSymbol> final_shape; in ProcessReduceNode() local
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | strided_slice_op.cc | 84 TensorShape final_shape; in EmitDynamicSlice() local
282 TensorShape final_shape; in Compile() local
430 PartialTensorShape processing_shape, final_shape; in CompileAsDynamicUpdateSlice() local
516 TensorShape processing_shape, final_shape; in Compile() local
656 TensorShape final_shape; in Compile() local
|
| H A D | reduction_ops_common.cc | 94 std::vector<int64_t> final_shape; in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | strided_slice_op.cc | 97 TensorShape processing_shape, final_shape; in Compute() local
205 TensorShape processing_shape, final_shape; in Compute() local
300 TensorShape final_shape; // After reshaping input. in Compute() local
|
| H A D | strided_slice_op_test.cc | 106 TensorShape processing_shape, final_shape; in BM_ValidateStridedSliceOp() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | PixelShuffle.cpp | 109 std::vector<int64_t> final_shape(self.sizes().begin(), self_sizes_batch_end); in math_pixel_shuffle() local
150 std::vector<int64_t> final_shape(self.sizes().begin(), self_sizes_batch_end); in math_pixel_unshuffle() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | strided_slice_op.cc | 175 PartialTensorShape* processing_shape, PartialTensorShape* final_shape, in ValidateStridedSliceOp()
427 TensorShape* processing_shape, TensorShape* final_shape, bool* is_identity, in ValidateStridedSliceOp()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| H A D | slice_ops.cc | 55 std::optional<nvinfer1::Dims> final_shape, std::optional<int> op_instance, in ConvertStridedSliceHelper()
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ |
| H A D | collectives.cc | 576 llvm::SmallVector<int64_t, 4> final_shape(input_tensor_shape.begin(), in EmitHaloExchange() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/clusters/ |
| H A D | single_machine_test.cc | 178 auto final_shape = ops::Shape(root.WithOpName("final_shape"), reshaped); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | tpu_space_to_depth_pass.cc | 258 SmallVector<int64_t, 4> final_shape = { in HandleConv2DFilter() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | array_ops.cc | 1773 PartialTensorShape processing_shape, final_shape; in __anon38bbb0e82502() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes.cc | 2575 PartialTensorShape final_shape; in ConvertSlice() local
2702 PartialTensorShape final_shape; in ConvertStridedSlice() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | arithmetic_optimizer.cc | 3951 TensorShape final_shape; in GetStridedSliceAxis() local
|