| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 44 inline void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
71 inline void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
87 void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
103 void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
113 inline void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
148 inline void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
168 void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
191 void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
208 inline void Conv(const float* input_data, const Dims<4>& input_dims, in Conv()
234 void Conv(const float* input_data, const Dims<4>& input_dims, in Conv()
[all …]
|
| D | reduce.h | 55 inline bool Reduce(const In* input_data, const int* input_dims, in Reduce()
80 inline bool Reduce(const In* input_data, const int* input_dims, in Reduce()
144 inline bool ReduceSumImpl(const In* input_data, const int* input_dims, in ReduceSumImpl()
180 inline bool ReduceGeneric(const T* input_data, const int* input_dims, in ReduceGeneric()
217 inline bool Mean(const T* input_data, const int* input_dims, in Mean()
379 float input_scale, const int* input_dims, in QuantizedMeanOrSum()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_benchmark.h | 46 void SpatialConvolution(Dimensions input_dims, Dimensions filter_dims) { in SpatialConvolution()
76 void SpatialConvolutionBackwardInput(Dimensions input_dims, in SpatialConvolutionBackwardInput()
115 void SpatialConvolutionBackwardKernel(Dimensions input_dims, in SpatialConvolutionBackwardKernel()
177 void CuboidConvolution(Dimensions input_dims, Dimensions filter_dims) { in CuboidConvolution()
208 void CuboidConvolutionBackwardInput(Dimensions input_dims, in CuboidConvolutionBackwardInput()
249 void CuboidConvolutionBackwardKernel(Dimensions input_dims, in CuboidConvolutionBackwardKernel()
|
| D | reduce_join_op.cc | 78 int32_t input_dims) { in GetReducedIndices()
93 int32_t input_dims, in MakeUnreducedIndices()
128 const int32_t input_dims = input_shape.dims(); in Compute() local
|
| D | tile_ops.cc | 175 const int input_dims = input.dims(); in Compute() local
341 const int input_dims = input.dims(); in Compute() local
421 const std::vector<Tmultiples>& input_dims, in HandleCaseImpl()
514 OpKernelContext* context, const std::vector<Tmultiples>& input_dims, in HandleCase()
|
| D | sparse_slice_op.cc | 40 const int input_dims = input_shape.NumElements(); in operator ()() local
126 const int input_dims = input_shape.NumElements(); in SparseSliceOpImpl() local
|
| D | where_op.cc | 141 const int input_dims = input.dims(); in Compute() local
263 const int input_dims = input.dims(); in ComputeAsync() local
273 void ComputeAsyncType(const Tensor& input, const int input_dims, in ComputeAsyncType()
|
| D | eigen_benchmark_cpu_test.cc | 42 typename Benchmark::Dimensions input_dims(input_batches, input_height, in SpatialConvolution() local
70 typename Benchmark::Dimensions input_dims(input_batches, input_height, in SpatialConvolutionBackwardInput() local
97 typename Benchmark::Dimensions input_dims(input_batches, input_height, in SpatialConvolutionBackwardKernel() local
260 typename Benchmark::Dimensions input_dims( in CuboidConvolution() local
289 typename Benchmark::Dimensions input_dims( in CuboidConvolutionBackwardInput() local
317 typename Benchmark::Dimensions input_dims( in CuboidConvolutionBackwardKernel() local
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_convolution_sycl.cpp | 44 Eigen::array<IndexType, 3> input_dims = {{indim0, indim1, indim2}}; in test_larg_expr1D() local
105 Eigen::array<IndexType, 3> input_dims = {{indim0, indim1, indim2}}; in test_larg_expr2D() local
166 Eigen::array<IndexType, 3> input_dims = {{indim0, indim1, indim2}}; in test_larg_expr3D() local
221 Eigen::array<IndexType, 2> input_dims = {{3, 3}}; in test_evals() local
267 Eigen::array<IndexType, 2> input_dims = {{3, 3}}; in test_expr() local
316 Eigen::array<IndexType, 1> input_dims = {{3}}; in test_modes() local
405 Eigen::array<IndexType, 1> input_dims = {{13}}; in test_strides() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 139 inline void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
171 inline void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
187 void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
203 void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
281 inline void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
321 inline void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
341 void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
364 void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
685 inline void FullyConnected(const float* input_data, const Dims<4>& input_dims, in FullyConnected()
704 void FullyConnected(const float* input_data, const Dims<4>& input_dims, in FullyConnected()
[all …]
|
| D | reduce.h | 312 void ReduceIsCopy(const T* input_data, const int* input_dims, in ReduceIsCopy()
328 const int* input_dims, U* output_data, in ReduceImpl()
387 inline bool Reduce(const In* input_data, const int* input_dims, in Reduce()
403 float input_scale, const int* input_dims, in QuantizedMeanOrSum()
495 inline bool ReduceDispatcher(const T* input_data, const int* input_dims, in ReduceDispatcher()
671 inline bool MeanGeneral(const T* input_data, const int* input_dims, in MeanGeneral()
729 inline bool Mean(const T* input_data, const int* input_dims, in Mean()
779 inline bool ReduceGeneric(const T* input_data, const int* input_dims, in ReduceGeneric()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | quantization_util_test.cc | 149 auto input_dims = BuildTfLiteIntArray({1, 3, 2, 1}); in TEST() local
174 auto input_dims = BuildTfLiteIntArray({1, 3, 2, 1}); in TEST() local
199 auto input_dims = BuildTfLiteIntArray({1, 3, 2, 1}); in TEST() local
221 auto input_dims = BuildTfLiteIntArray({1, 3, 2, 1}); in TEST() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convolution_test.cc | 260 std::vector<int64_t> input_dims = {1, 4, 2, 3, 3}; in XLA_TEST_F() local
325 std::vector<int64_t> input_dims = {1, 3, 3, 5}; in RunTest() local
382 std::vector<int64_t> input_dims = {1, 3, 3, 5}; in RunTest() local
446 std::vector<int64_t> input_dims = {1, 4, 4, 5}; in RunTest() local
514 std::vector<int64_t> input_dims = {1, 4, 4, 512}; in RunTest() local
576 std::vector<int64_t> input_dims = {1, 4, 4, 512}; in RunTest() local
643 std::vector<int64_t> input_dims = {256, 4, 4, 512}; in RunTest() local
709 std::vector<int64_t> input_dims = {256, 4, 4, 512}; in RunTest() local
777 std::vector<int64_t> input_dims = {1, 4, 4, 5}; in RunTest() local
850 std::vector<int64_t> input_dims = {1, 4, 4, 160}; in RunTest() local
[all …]
|
| /external/libtextclassifier/native/utils/tflite/ |
| D | blacklist_base.cc | 39 TfLiteIntArray* input_dims = op->GetInputShape(context, node); in Resize() local
56 TfLiteIntArray* input_dims = op->GetInputShape(context, node); in Eval() local
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | c_test.c | 93 int input_dims[1] = {2}; in TestInferenceUsingSignature() local
197 int input_dims[] = {sizes[i]}; in TestRepeatResizeInputTensor() local
252 int input_dims[1] = {2}; in TestInferenceUsingInterpreter() local
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEReduceMean.cpp | 47 const int input_dims = input->num_dimensions(); in validate_config() local
122 const int input_dims = tmp_input->info()->num_dimensions(); in configure() local
|
| /external/ComputeLibrary/src/gpu/cl/operators/ |
| D | ClWinogradConv2d.cpp | 51 Size2D winograd_output_tile(const Size2D &input_dims, const Size2D &kernel_dims, DataLayout data_la… in winograd_output_tile()
114 …const Size2D input_dims = Size2D(src->tensor_shape()[idx_width], src->tensor_shape()[idx_height]); in validate_arguments() local
185 …const Size2D input_dims = Size2D(src->tensor_shape()[idx_width], src->tensor_shape()[idx_height]); in configure() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | reorder_reshape_transpose.cc | 58 std::vector<int> ComputeNewPerm(std::vector<int> input_dims, in ComputeNewPerm()
171 std::vector<int> input_dims = input_shape.dims(); in Run() local
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLReduceMean.cpp | 51 const int input_dims = input->num_dimensions(); in validate_config() local
146 const int input_dims = tmp_input->info()->num_dimensions(); in configure() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | dynamic_update_slice.cc | 95 std::vector<int> ClampStartIndices(int input_dims, const int32_t* indices_data, in ClampStartIndices()
116 const int input_dims = input_shape.DimensionsCount(); in DynamicUpdateSlice() local
|
| D | mirror_pad.cc | 48 const TfLiteIntArray* input_dims = nullptr; member
85 const int input_dims = NumDimensions(input); in GetPaddedOutputShape() local
175 const int input_dims = NumDimensions(input_tensor); in Eval() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | broadcast.cc | 32 absl::Span<int64_t const> input_dims = input_shape.dimensions(); in BroadcastTo() local
|
| D | sorting.cc | 56 auto input_dims = input_shape.dimensions(); in TopK() local
159 auto input_dims = input_shape.dimensions(); in TopKWithPartitions() local
|
| /external/XNNPACK/test/ |
| D | constant-pad-operator-tester.h | 114 std::array<size_t, XNN_MAX_TENSOR_DIMS> input_dims; in TestX8() local
229 std::array<size_t, XNN_MAX_TENSOR_DIMS> input_dims; in TestX16() local
344 std::array<size_t, XNN_MAX_TENSOR_DIMS> input_dims; in TestX32() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/ |
| D | cwise_op_unary_benchmark.h | 115 std::array<ssize_t, rank> input_dims; in TestUnaryMlirBenchmark() local
150 std::array<ssize_t, rank> input_dims; in RunUnaryMlirBenchmark() local
190 std::array<ssize_t, rank> input_dims; in RunUnaryEigenBenchmark() local
|