| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | DepthwiseConvolutionLayerFixture.h | 293 …void setup(TensorShape in_shape, Size2D kernel_size, PadStrideInfo pad_stride_info, Size2D dilatio… in setup()
644 …void setup(TensorShape in_shape, Size2D kernel_size, PadStrideInfo pad_stride_info, Size2D dilatio… in setup()
658 …void setup(TensorShape in_shape, Size2D kernel_size, PadStrideInfo pad_stride_info, Size2D dilatio… in setup()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | bcast_ops.cc | 44 const TensorShape in_shape = ctx->InputShape(i); in Compile() local
97 const TensorShape in_shape = ctx->InputShape(i); in Compile() local
|
| H A D | concat_op.cc | 76 const TensorShape& in_shape = shapes[i]; in Compile() local
|
| H A D | lrn_ops.cc | 38 const TensorShape in_shape = ctx->InputShape(0); in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | image_grad_test_base.py | 132 in_shape, argument
235 def _gpuVsCpuCase(self, in_shape, out_shape, align_corners, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | scoped_allocator_ops_test.cc | 262 void BuildNodeDef(const TensorShape& in_shape, DataType dtype, in BuildNodeDef()
278 void MakeOp(const TensorShape& in_shape, DataType dtype, const string& name, in MakeOp()
|
| H A D | split_op_test.cc | 32 TensorShape in_shape(chunk_size); in MakeGraph() local
|
| H A D | split_v_op_test.cc | 63 TensorShape in_shape(total_size); in MakeGraph() local
|
| H A D | cudnn_pooling_gpu.cc | 42 const auto in_shape = tensor_in.shape(); in Compute() local
|
| H A D | fractional_avg_pool_op.cc | 315 TensorShape in_shape; in Compute() local
|
| H A D | bias_op.cc | 335 BiasAddParams(const SpatialArray& in_shape, TensorFormat data_format, in BiasAddParams()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/image_ops/ |
| H A D | extract_volume_patches_grad_test.py | 56 def testGradient(self, in_shape, ksizes, strides): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| H A D | convert_trivial_transpose_to_reshape.cc | 28 std::vector<int> in_shape) { in TransposeAffectsMemoryOrder()
|
| H A D | merge_reshape_into_preceding_transpose.cc | 66 std::vector<int> in_shape = input_array.shape().dims(); in ReshapeToTranspose() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | schedule_aware_collective_ops_cse.cc | 35 const Shape& in_shape = inst->operand(0)->shape(); in IsAddingOnlyDegenerateDimensions() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| H A D | cudnn_deterministic_base.py | 77 def _random_out_op(self, in_shape, filter_shape, strides, padding, dilations): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | PadNd.cpp | 110 const auto in_shape = self.sym_sizes(); in _pad_circular_symint() local
|
| H A D | Sorting.cpp | 281 std::vector<int64_t> in_shape(out_shape.size()); in quantile_compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/ |
| H A D | tf_jitrt_symbolic_shape_optimization.cc | 186 auto in_shape = analysis.GetShapeInfo(from); in isNonExpandingBroadcast() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/NEON/ |
| H A D | GEMM.cpp | 77 const TensorShape in_shape(dim0_value, dim1_value); in validate_zero_padding() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | copy_test.cc | 243 Shape in_shape = ShapeUtil::MakeShapeWithLayout(F32, {0, 0}, {0, 1}); in XLA_TEST_F() local
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/operators/ |
| H A D | CpuWinogradConv2d.cpp | 91 Tensor4DShape in_shape{ internal_get_shape(src) }; in get_winograd_kernel_implementation() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_graph.cc | 408 tensorflow::TensorShapeProto in_shape = ComputeTRTNodeIOShape( in CreateTRTNode() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_dataflow_analysis_test.cc | 2237 Shape in_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
2353 Shape in_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/nccl/ |
| H A D | nccl_manager_test.cc | 166 TensorShape in_shape, TensorShape out_shape) { in MakeGatherTestCase()
|