| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 52 const Dims<4>& output_dims) { in DepthwiseConv()
78 const Dims<4>& output_dims) { in DepthwiseConv()
92 const Dims<4>& output_dims) { in DepthwiseConv()
107 float* output_data, const Dims<4>& output_dims) { in DepthwiseConv()
123 const Dims<4>& output_dims) { in DepthwiseConv()
157 const Dims<4>& output_dims) { in DepthwiseConv()
176 uint8* output_data, const Dims<4>& output_dims) { in DepthwiseConv()
199 const Dims<4>& output_dims) { in DepthwiseConv()
214 float* output_data, const Dims<4>& output_dims, in Conv()
239 float* output_data, const Dims<4>& output_dims, float* im2col_data, in Conv()
[all …]
|
| D | reduce.h | 56 const int* output_dims, const int input_num_dims, in Reduce()
81 const int* output_dims, const int input_num_dims, in Reduce()
145 const int* output_dims, const int input_num_dims, in ReduceSumImpl()
182 const int* output_dims, const int output_num_dims, in ReduceGeneric()
219 const int* output_dims, const int output_num_dims, in Mean()
382 const int* output_dims, in QuantizedMeanOrSum()
|
| /external/XNNPACK/test/ |
| D | static-transpose.cc | 52 std::vector<size_t> output_dims = PermuteInputDimensions(dims, perm); in TEST_F() local
100 std::vector<size_t> output_dims = PermuteInputDimensions(dims, perm); in TEST_F() local
143 std::vector<size_t> output_dims = PermuteInputDimensions(dims, perm); in TEST_F() local
193 std::vector<size_t> output_dims = PermuteInputDimensions(dims, perm); in TEST_F() local
258 std::vector<size_t> output_dims = PermuteInputDimensions(dims, perm); in TEST_F() local
319 std::vector<size_t> output_dims = PermuteInputDimensions(dims, perm); in TEST_F() local
|
| D | static-reshape.cc | 144 std::vector<size_t> output_dims = dims; in TEST_F() local
207 std::vector<size_t> output_dims = dims; in TEST_F() local
268 std::vector<size_t> output_dims = dims; in TEST_F() local
|
| D | static-constant-pad.cc | 191 std::vector<size_t> output_dims = dims; in TEST_F() local
265 std::vector<size_t> output_dims = dims; in TEST_F() local
337 std::vector<size_t> output_dims = dims; in TEST_F() local
|
| D | constant-pad-operator-tester.h | 117 std::array<size_t, XNN_MAX_TENSOR_DIMS> output_dims; in TestX8() local
232 std::array<size_t, XNN_MAX_TENSOR_DIMS> output_dims; in TestX16() local
347 std::array<size_t, XNN_MAX_TENSOR_DIMS> output_dims; in TestX32() local
|
| D | binary-elementwise-operator-tester.h | 216 std::array<size_t, XNN_MAX_TENSOR_DIMS> output_dims; in TestQS8() local
398 std::array<size_t, XNN_MAX_TENSOR_DIMS> output_dims; in TestQU8() local
573 std::array<size_t, XNN_MAX_TENSOR_DIMS> output_dims; in TestF16() local
794 std::array<size_t, XNN_MAX_TENSOR_DIMS> output_dims; in TestF32() local
|
| D | subgraph-binary-tester.h | 114 std::array<size_t, XNN_MAX_TENSOR_DIMS> output_dims; variable
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 147 const Dims<4>& output_dims) { in DepthwiseConv()
178 const Dims<4>& output_dims) { in DepthwiseConv()
192 const Dims<4>& output_dims) { in DepthwiseConv()
207 float* output_data, const Dims<4>& output_dims) { in DepthwiseConv()
291 const Dims<4>& output_dims) { in DepthwiseConv()
330 const Dims<4>& output_dims) { in DepthwiseConv()
349 uint8* output_data, const Dims<4>& output_dims) { in DepthwiseConv()
372 const Dims<4>& output_dims) { in DepthwiseConv()
691 const Dims<4>& output_dims) { in FullyConnected()
707 float* output_data, const Dims<4>& output_dims) { in FullyConnected()
[all …]
|
| D | reduce.h | 406 const int* output_dims, const int output_num_dims, in QuantizedMeanOrSum()
496 const int input_num_dims, const int* output_dims, in ReduceDispatcher()
673 const int* output_dims, const int output_num_dims, in MeanGeneral()
731 const int* output_dims, const int output_num_dims, in Mean()
781 const int* output_dims, const int output_num_dims, in ReduceGeneric()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | indexed_array_analysis.h | 191 absl::Span<const int64_t> output_dims() const { return output_dims_; } in output_dims() function
196 std::vector<int64_t> output_dims, Shape shape) in ScalarIndexedArray()
227 std::vector<int64_t> output_dims, in ScalarIndexedConstantArray()
339 std::vector<int64_t> output_dims, Shape shape) { in ConstructScalarIndexedArray()
|
| D | gather_simplifier.cc | 62 std::vector<int64_t> output_dims = {start_indices->shape().dimensions(0)}; in ExpandInstruction() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | broadcast.cc | 33 absl::Span<int64_t const> output_dims) { in BroadcastTo()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | broadcast.cc | 29 absl::Span<int64_t const> output_dims) { in BroadcastTo()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | one_hot_op.cc | 39 const int output_dims = indices_dims + 1; in Compile() local
|
| D | tile_ops.cc | 72 std::vector<int64_t> output_dims(input_shape.dims()); in Compile() local
|
| /external/tensorflow/tensorflow/lite/java/src/test/native/ |
| D | interpreter_test_jni.cc | 40 TfLiteIntArray* output_dims = TfLiteIntArrayCopy(input->dims); in Java_org_tensorflow_lite_InterpreterTest_getNativeHandleForDelegate() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| D | xla_ops.cc | 259 std::vector<shape_inference::DimensionHandle> output_dims; in XlaDotShapeFunction() local
525 std::vector<shape_inference::DimensionHandle> output_dims; in __anonb720852f0302() local
760 std::vector<shape_inference::DimensionHandle> output_dims; in __anonb720852f0902() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | conv_depthwise_common.h | 41 std::vector<int64_t> output_dims; member
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | convert_trivial_transpose_to_reshape.cc | 80 std::vector<int> const& output_dims = output_array.shape().dims(); in Run() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | redux_functor.h | 49 auto output_dims = output->template flat<OutputT>().dimensions(); in operator() local
218 auto output_dims = output->template flat<OutputT>().dimensions(); in operator() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | exp.cc | 45 TfLiteIntArray* output_dims = TfLiteIntArrayCopy(op_context.input->dims); in Prepare() local
|
| D | fake_quant.cc | 58 TfLiteIntArray* output_dims = TfLiteIntArrayCopy(op_context.input->dims); in Prepare() local
|
| D | where.cc | 51 TfLiteIntArray* output_dims = TfLiteIntArrayCreate(2); in ResizeOutputTensor() local
|
| /external/libtextclassifier/native/utils/tflite/ |
| D | blacklist_base.cc | 40 TfLiteIntArray* output_dims = TfLiteIntArrayCreate(input_dims->size + 1); in Resize() local
|