| /external/ComputeLibrary/src/gpu/cl/kernels/ |
| D | ClMulKernel.cpp | 70 …const TensorShape &out_shape = TensorShape::broadcast_shape(src1->tensor_shape(), src2->tensor_sha… in validate_arguments() local
122 …const TensorShape &out_shape = TensorShape::broadcast_shape(src1->tensor_shape(), src2->tensor_sha… in configure() local
282 const TensorShape &out_shape = dst->info()->tensor_shape(); in run_op() local
333 …const TensorShape &out_shape = TensorShape::broadcast_shape(src1->tensor_shape(), src2->tensor_sha… in validate_arguments_complex() local
362 …const TensorShape &out_shape = TensorShape::broadcast_shape(src1->tensor_shape(), src2->tensor_sha… in configure() local
402 const TensorShape &out_shape = dst->info()->tensor_shape(); in run_op() local
|
| D | ClElementwiseKernel.cpp | 78 …t ITensorInfo &src1, const ITensorInfo &src2, const ITensorInfo &dst, const TensorShape &out_shape) in validate_in_place_output_shape()
104 …const TensorShape out_shape = TensorShape::broadcast_shape(src1.tensor_shape(), src2.tensor_shape(… in validate_arguments_with_float_only_supported_rules() local
130 …const TensorShape out_shape = TensorShape::broadcast_shape(src1->tensor_shape(), src2->tensor_shap… in validate_arguments_divide_operation() local
165 …const TensorShape out_shape = TensorShape::broadcast_shape(src1.tensor_shape(), src2.tensor_shape(… in validate_arguments_with_arithmetic_rules() local
230 const TensorShape &out_shape = broadcast_pair.first; in validate_and_configure_window_for_arithmetic_operators() local
240 const TensorShape &out_shape = broadcast_pair.first; in validate_and_configure_window_for_logical_binary_operators() local
251 const TensorShape &out_shape = broadcast_pair.first; in validate_and_configure_window_for_division() local
306 const TensorShape &out_shape = dst->info()->tensor_shape(); in run_op() local
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEReduceMean.cpp | 60 TensorShape out_shape = input->tensor_shape(); in validate_config() local
129 …TensorShape out_shape = i == 0 ? tmp_input->info()->tensor_shape() : (&_reduced_outs[i - 1])->info… in configure() local
153 TensorShape out_shape = tmp_input->info()->tensor_shape(); in configure() local
|
| D | NECropResize.cpp | 53 …TensorShape out_shape(input->tensor_shape()[0], crop_size.x, crop_size.y, boxes->tensor_shape()[1]… in validate() local
67 TensorShape out_shape(input->info()->tensor_shape()[0], crop_size.x, crop_size.y); in configure() local
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLReduceMean.cpp | 64 TensorShape out_shape = input->tensor_shape(); in validate_config() local
153 …TensorShape out_shape = i == 0 ? tmp_input->info()->tensor_shape() : (&_reduced_outs[i - 1])->info… in configure() local
178 TensorShape out_shape = tmp_input->info()->tensor_shape(); in configure() local
|
| D | CLCropResize.cpp | 55 …const TensorShape out_shape(input->info()->tensor_shape()[0], static_cast<uint32_t>(abs(end[0] - s… in configure_crop() local
80 …TensorShape out_shape(input->tensor_shape()[0], crop_size.x, crop_size.y, boxes->tensor_shape()[1]… in validate() local
103 TensorShape out_shape(input->info()->tensor_shape()[0], crop_size.x, crop_size.y); in configure() local
|
| /external/ComputeLibrary/tests/validation/reference/ |
| D | CropResize.cpp | 37 SimpleTensor<float> scale_image(const SimpleTensor<float> &in, const TensorShape &out_shape, Interp… in scale_image()
126 …TensorShape out_shape(src.shape()[0], static_cast<uint32_t>(abs(end[0] - start[0])) + 1, static_ca… in crop_image() local
166 const TensorShape out_shape(src.shape()[0], crop_size.x, crop_size.y, boxes.shape()[1]); in crop_and_resize() local
|
| D | Unstack.cpp | 67 TensorShape out_shape = input_tensor.shape(); in get_slice() local
|
| /external/ComputeLibrary/src/core/CL/kernels/ |
| D | CLComparisonKernel.cpp | 62 …const TensorShape out_shape = TensorShape::broadcast_shape(input1.tensor_shape(), input2.tensor_sh… in validate_arguments() local
78 …const TensorShape &out_shape = TensorShape::broadcast_shape(input1.tensor_… in validate_and_configure_window() local
179 const TensorShape &out_shape = _output->info()->tensor_shape(); in run() local
|
| /external/ComputeLibrary/src/cpu/kernels/ |
| D | CpuSubKernel.cpp | 121 …const TensorShape out_shape = TensorShape::broadcast_shape(src0.tensor_shape(), src1.tensor_shape(… in validate_arguments() local
143 …const TensorShape &out_shape = TensorShape::broadcast_shape(src0->tensor_shape(), src1->tensor_sha… in configure() local
|
| D | CpuAddKernel.cpp | 189 …const TensorShape out_shape = TensorShape::broadcast_shape(src0.tensor_shape(), src1.tensor_shape(… in validate_arguments() local
229 …const TensorShape &out_shape = TensorShape::broadcast_shape(src0->tensor_shape(), src1->tensor_sha… in configure() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | batch_ops.cc | 105 shape_inference::ShapeHandle out_shape; in __anon9f94a71a0202() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | diag_op.cc | 53 TensorShape out_shape; in Compute() local
93 TensorShape out_shape; in Compute() local
|
| D | decode_padded_raw_op.cc | 64 TensorShape out_shape = input.shape(); in Compute() local
|
| D | maxpooling_op.cc | 846 TensorShape out_shape({params.tensor_in_batch, params.out_height, in Compute() local
929 TensorShape out_shape({params.tensor_in_batch, params.out_height, in Compute() local
1001 TensorShape out_shape({params.tensor_in_batch, params.out_height, in Compute() local
1139 TensorShape out_shape({params.tensor_in_batch, params.tensor_in_rows, in Compute() local
1207 TensorShape out_shape({params.tensor_in_batch, params.out_height, in Compute() local
1267 TensorShape out_shape = in Compute() local
1402 TensorShape out_shape = in Compute() local
|
| D | decode_raw_op.cc | 59 TensorShape out_shape = input.shape(); in Compute() local
|
| /external/ComputeLibrary/src/graph/mutators/ |
| D | InPlaceOperationMutator.cpp | 126 const auto out_shape = current_output_tensor->desc().shape; in try_in_place_depthwiseconv() local
171 const TensorShape out_shape = TensorShape::broadcast_shape(shape0, shape1); in try_in_place_elementwise() local
|
| /external/ComputeLibrary/src/core/helpers/ |
| D | WindowHelpers.h | 213 const TensorShape out_shape = TensorShape::broadcast_shape(shapes...); in compute_output_shape_and_window() local
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_argmax_gpu.cu | 78 array<DenseIndex, 3> out_shape; in test_gpu_argmax_dim() local
167 array<DenseIndex, 3> out_shape; in test_gpu_argmin_dim() local
|
| D | cxx11_tensor_argmax_sycl.cpp | 81 array<DenseIndex, 3> out_shape; in test_sycl_argmax_dim() local
169 array<DenseIndex, 3> out_shape; in test_sycl_argmin_dim() local
|
| /external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/components/cl/ |
| D | ClComponentElementwiseBinary.cpp | 62 …const TensorShape out_shape = TensorShape::broadcast_shape(lhs->tensor_shape(), rhs->tensor_shape(… in validate() local
|
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NELogicalKernel.cpp | 281 TensorShape out_shape = input1->tensor_shape(); in configure() local
300 TensorShape out_shape = input1->tensor_shape(); in validate() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | merge_reshape_into_preceding_transpose.cc | 67 std::vector<int> out_shape = output_array.shape().dims(); in ReshapeToTranspose() local
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 346 const auto& out_shape = out->shape(); in ValidateAndInitializeToDense() local
380 const auto& out_shape = out->shape(); in ToDense() local
396 const auto& out_shape = out->shape().dim_sizes(); in ToDense() local
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | determinant_op.cc | 149 TensorShape out_shape; in ComputeAsync() local
289 TensorShape out_shape; in ComputeAsync() local
|