Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/core/kernels/
Deigen_benchmark.h45 void SpatialConvolution(Dimensions input_dims, Dimensions filter_dims) { in SpatialConvolution()
77 void SpatialConvolutionBackwardInput(Dimensions input_dims, in SpatialConvolutionBackwardInput()
118 void SpatialConvolutionBackwardKernel(Dimensions input_dims, in SpatialConvolutionBackwardKernel()
180 void CuboidConvolution(Dimensions input_dims, Dimensions filter_dims) { in CuboidConvolution()
213 void CuboidConvolutionBackwardInput(Dimensions input_dims, in CuboidConvolutionBackwardInput()
256 void CuboidConvolutionBackwardKernel(Dimensions input_dims, in CuboidConvolutionBackwardKernel()
Dreduce_join_op.cc77 int32 input_dims) { in GetReducedIndices()
92 int32 input_dims, in MakeUnreducedIndices()
127 const int32 input_dims = input_shape.dims(); in Compute() local
Dwhere_op.cc141 const int input_dims = input.dims(); in Compute() local
268 const int input_dims = input.dims(); in ComputeAsync() local
278 void ComputeAsyncType(const Tensor& input, const int input_dims, in ComputeAsyncType()
Dtile_ops.cc197 const int input_dims = input.dims(); in Compute() local
372 const int input_dims = input.dims(); in Compute() local
452 const std::vector<Tmultiples>& input_dims, in HandleCaseImpl()
545 OpKernelContext* context, const std::vector<Tmultiples>& input_dims, in HandleCase()
Deigen_benchmark_cpu_test.cc44 typename Benchmark::Dimensions input_dims(input_batches, input_height, in SpatialConvolution() local
73 typename Benchmark::Dimensions input_dims(input_batches, input_height, in SpatialConvolutionBackwardInput() local
101 typename Benchmark::Dimensions input_dims(input_batches, input_height, in SpatialConvolutionBackwardKernel() local
266 typename Benchmark::Dimensions input_dims( in CuboidConvolution() local
296 typename Benchmark::Dimensions input_dims( in CuboidConvolutionBackwardInput() local
325 typename Benchmark::Dimensions input_dims( in CuboidConvolutionBackwardKernel() local
Dsparse_slice_op.cc58 const int input_dims = input_shape.NumElements(); in Compute() local
/external/tensorflow/tensorflow/lite/micro/kernels/
Darg_min_max_test.cc77 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestArgMinMaxFloat() local
103 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestArgMinMaxQuantized() local
130 const int input_dims[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
144 const int input_dims[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
159 const int input_dims[] = {4, 1, 1, 1, input_size}; in TF_LITE_MICRO_TEST() local
178 const int input_dims[] = {4, 1, 1, 1, input_size}; in TF_LITE_MICRO_TEST() local
197 const int input_dims[] = {4, 1, 1, 1, input_size}; in TF_LITE_MICRO_TEST() local
216 const int input_dims[] = {4, 1, 1, 1, input_size}; in TF_LITE_MICRO_TEST() local
235 const int input_dims[] = {4, 1, 1, 2, 4}; in TF_LITE_MICRO_TEST() local
254 const int input_dims[] = {4, 1, 1, 2, 4}; in TF_LITE_MICRO_TEST() local
[all …]
Dceil_test.cc28 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestCeil() local
75 const int input_dims[] = {1, 2}; in TF_LITE_MICRO_TEST() local
83 const int input_dims[] = {4, 2, 1, 1, 5}; in TF_LITE_MICRO_TEST() local
Dround_test.cc28 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestRound() local
74 const int input_dims[] = {1, 6}; in TF_LITE_MICRO_TEST() local
82 const int input_dims[] = {4, 2, 1, 1, 6}; in TF_LITE_MICRO_TEST() local
Dpad_test.cc116 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestPadFloat() local
142 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestPadV2Float() local
173 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestPadQuantized() local
209 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestPadV2Quantized() local
249 const int input_dims[] = {4, 1, 2, 2, 1}; in TF_LITE_MICRO_TEST() local
263 const int input_dims[] = {4, 1, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
282 const int input_dims[] = {4, 1, 2, 2, 1}; in TF_LITE_MICRO_TEST() local
298 const int input_dims[] = {4, 1, 2, 2, 1}; in TF_LITE_MICRO_TEST() local
320 const int input_dims[] = {4, 1, 2, 2, 1}; in TF_LITE_MICRO_TEST() local
346 const int input_dims[] = {4, 1, 2, 2, 1}; in TF_LITE_MICRO_TEST() local
[all …]
Dreshape_test.cc111 TfLiteIntArray* input_dims = IntArrayFromInitializer(input_dims_data); in TestReshape() local
189 TfLiteIntArray* input_dims = IntArrayFromInitializer({3, 1, 2, 2}); in TF_LITE_MICRO_TEST() local
253 TfLiteIntArray* input_dims = IntArrayFromInitializer({1, 1}); in TF_LITE_MICRO_TEST() local
Dactivations_test.cc29 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestReluFloat() local
85 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestRelu6Float() local
144 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestReluUint8() local
209 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestRelu6Uint8() local
274 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestReluInt8() local
340 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestRelu6Int8() local
Dsoftmax_test.cc31 TfLiteIntArray* input_dims = IntArrayFromInitializer(input_dims_data); in TestSoftmaxFloat() local
95 TfLiteIntArray* input_dims = IntArrayFromInitializer(input_dims_data); in TestSoftmaxQuantized() local
161 TfLiteIntArray* input_dims = IntArrayFromInitializer(input_dims_data); in TestSoftmaxQuantizedSigned() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h41 inline void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
68 inline void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
84 void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
100 void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
110 inline void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
145 inline void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
165 void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
188 void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
205 inline void Conv(const float* input_data, const Dims<4>& input_dims, in Conv()
231 void Conv(const float* input_data, const Dims<4>& input_dims, in Conv()
[all …]
Dreduce.h31 inline bool Reduce(const In* input_data, const int* input_dims, in Reduce()
87 inline bool ReduceSumImpl(const In* input_data, const int* input_dims, in ReduceSumImpl()
122 inline bool ReduceGeneric(const T* input_data, const int* input_dims, in ReduceGeneric()
151 inline bool Mean(const T* input_data, const int* input_dims, in Mean()
313 float input_scale, const int* input_dims, in QuantizedMeanOrSum()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h139 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()
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/compiler/xla/tests/
Dconvolution_test.cc413 std::vector<int64> input_dims = {1, 4, 2, 3, 3}; in XLA_TEST_F() local
479 std::vector<int64> input_dims = {1, 3, 3, 5}; in RunTest() local
538 std::vector<int64> input_dims = {1, 3, 3, 5}; in RunTest() local
604 std::vector<int64> input_dims = {1, 4, 4, 5}; in RunTest() local
674 std::vector<int64> input_dims = {1, 4, 4, 512}; in RunTest() local
738 std::vector<int64> input_dims = {1, 4, 4, 512}; in RunTest() local
807 std::vector<int64> input_dims = {256, 4, 4, 512}; in RunTest() local
876 std::vector<int64> input_dims = {256, 4, 4, 512}; in RunTest() local
947 std::vector<int64> input_dims = {1, 4, 4, 5}; in RunTest() local
1022 std::vector<int64> input_dims = {1, 4, 4, 160}; in RunTest() local
[all …]
/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/xla/client/lib/
Dsorting.cc40 auto input_dims = input_shape.dimensions(); in TopK() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtile_ops.cc56 const int input_dims = input_shape.dims(); in Compile() local
Dslice_op.cc56 const int input_dims = input_shape.dims(); in Compile() local
Droll_op.cc34 int64 input_dims = input_shape.dims(); in Compile() local
Dindex_ops.cc50 const int input_dims = input_shape.dims(); in Compile() local
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Dbroadcast.cc35 absl::Span<int64 const> input_dims = in BroadcastTo() local
/external/tensorflow/tensorflow/lite/kernels/
Dmirror_pad.cc44 const TfLiteIntArray* input_dims = nullptr; member
81 const int input_dims = NumDimensions(input); in GetPaddedOutputShape() local
169 const int input_dims = NumDimensions(input_tensor); in Eval() local

1234