Home
last modified time | relevance | path

Searched defs:input_dims (Results 1 – 25 of 90) sorted by relevance

1234

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h42 inline void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
69 inline void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
85 void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
101 void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
111 inline void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
146 inline void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
166 void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
189 void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
206 inline void Conv(const float* input_data, const Dims<4>& input_dims, in Conv()
232 void Conv(const float* input_data, const Dims<4>& input_dims, in Conv()
[all …]
Dreduce.h53 inline bool Reduce(const In* input_data, const int* input_dims, in Reduce()
78 inline bool Reduce(const In* input_data, const int* input_dims, in Reduce()
142 inline bool ReduceSumImpl(const In* input_data, const int* input_dims, in ReduceSumImpl()
178 inline bool ReduceGeneric(const T* input_data, const int* input_dims, in ReduceGeneric()
215 inline bool Mean(const T* input_data, const int* input_dims, in Mean()
377 float input_scale, const int* input_dims, in QuantizedMeanOrSum()
/external/tensorflow/tensorflow/core/kernels/
Deigen_benchmark.h46 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()
Dreduce_join_op.cc77 int32_t input_dims) { in GetReducedIndices()
92 int32_t input_dims, in MakeUnreducedIndices()
127 const int32_t input_dims = input_shape.dims(); in Compute() local
Dtile_ops.cc175 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()
Dwhere_op.cc141 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()
Deigen_benchmark_cpu_test.cc42 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
Dsparse_slice_op.cc58 const int input_dims = input_shape.NumElements(); in Compute() local
/external/eigen/unsupported/test/
Dcxx11_tensor_convolution_sycl.cpp44 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/ComputeLibrary/src/runtime/CL/functions/
DCLWinogradConvolutionLayer.cpp45 Size2D winograd_output_tile(const Size2D &input_dims, const Size2D &kernel_dims, DataLayout data_la… in winograd_output_tile()
124 …const Size2D input_dims = Size2D(input->info()->tensor_shape()[idx_width], input->info()->tensor_… in configure() local
176 …const Size2D input_dims = Size2D(input->tensor_shape()[idx_width], input->tensor_shape()[idx_heig… in validate() local
DCLReduceMean.cpp49 const int input_dims = input->num_dimensions(); in validate_config() local
142 const int input_dims = tmp_input->info()->num_dimensions(); in configure() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h137 inline void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
169 inline void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
185 void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
201 void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
279 inline void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
319 inline void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
339 void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
362 void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
683 inline void FullyConnected(const float* input_data, const Dims<4>& input_dims, in FullyConnected()
702 void FullyConnected(const float* input_data, const Dims<4>& input_dims, in FullyConnected()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dquantization_util_test.cc149 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/
Dconvolution_test.cc265 std::vector<int64> input_dims = {1, 4, 2, 3, 3}; in XLA_TEST_F() local
331 std::vector<int64> input_dims = {1, 3, 3, 5}; in RunTest() local
390 std::vector<int64> input_dims = {1, 3, 3, 5}; in RunTest() local
456 std::vector<int64> input_dims = {1, 4, 4, 5}; in RunTest() local
526 std::vector<int64> input_dims = {1, 4, 4, 512}; in RunTest() local
590 std::vector<int64> input_dims = {1, 4, 4, 512}; in RunTest() local
659 std::vector<int64> input_dims = {256, 4, 4, 512}; in RunTest() local
728 std::vector<int64> input_dims = {256, 4, 4, 512}; in RunTest() local
799 std::vector<int64> input_dims = {1, 4, 4, 5}; in RunTest() local
874 std::vector<int64> input_dims = {1, 4, 4, 160}; in RunTest() local
[all …]
/external/libtextclassifier/native/utils/tflite/
Dblacklist_base.cc39 TfLiteIntArray* input_dims = op->GetInputShape(context, node); in Resize() local
56 TfLiteIntArray* input_dims = op->GetInputShape(context, node); in Eval() local
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEReduceMean.cpp48 const int input_dims = input->num_dimensions(); in validate_config() local
143 const int input_dims = tmp_input->info()->num_dimensions(); in configure() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dreorder_reshape_transpose.cc58 std::vector<int> ComputeNewPerm(std::vector<int> input_dims, in ComputeNewPerm()
171 std::vector<int> input_dims = input_shape.dims(); in Run() local
/external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/
Dcwise_op_unary_benchmark.h94 std::array<ssize_t, rank> input_dims; in TestUnaryMlirBenchmark() local
122 std::array<ssize_t, rank> input_dims; in RunUnaryMlirBenchmark() local
154 std::array<ssize_t, rank> input_dims; in RunUnaryEigenBenchmark() local
/external/tensorflow/tensorflow/lite/kernels/
Dmirror_pad.cc48 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/
Dsorting.cc57 auto input_dims = input_shape.dimensions(); in TopK() local
86 auto input_dims = input_shape.dimensions(); in TopKWithPartitions() local
/external/XNNPACK/test/
Dconstant-pad-operator-tester.h114 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/tf2xla/lib/
Dbroadcast.cc35 absl::Span<int64 const> input_dims = in BroadcastTo() local
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dconcatenation_op_builder.cc67 int input_dims = context->tensors[node->inputs->data[0]].dims->size; in IsConcatenationOpSupported() local
/external/tensorflow/tensorflow/lite/c/
Dc_api_test.cc53 std::array<int, 1> input_dims = {2}; in TEST() local
115 const std::array<int, 1> input_dims = {2}; in TEST() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dindex_ops.cc50 const int input_dims = input_shape.dims(); in Compile() local

1234