| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 47 const Dims<4>& output_dims) { in DepthwiseConv()
73 const Dims<4>& output_dims) { in DepthwiseConv()
87 const Dims<4>& output_dims) { in DepthwiseConv()
102 float* output_data, const Dims<4>& output_dims) { in DepthwiseConv()
118 const Dims<4>& output_dims) { in DepthwiseConv()
152 const Dims<4>& output_dims) { in DepthwiseConv()
171 uint8* output_data, const Dims<4>& output_dims) { in DepthwiseConv()
194 const Dims<4>& output_dims) { in DepthwiseConv()
209 float* output_data, const Dims<4>& output_dims, in Conv()
234 float* output_data, const Dims<4>& output_dims, float* im2col_data, in Conv()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 146 const Dims<4>& output_dims) { in DepthwiseConv()
172 const Dims<4>& output_dims) { in DepthwiseConv()
186 const Dims<4>& output_dims) { in DepthwiseConv()
201 float* output_data, const Dims<4>& output_dims) { in DepthwiseConv()
217 const Dims<4>& output_dims) { in DepthwiseConv()
251 const Dims<4>& output_dims) { in DepthwiseConv()
270 uint8* output_data, const Dims<4>& output_dims) { in DepthwiseConv()
293 const Dims<4>& output_dims) { in DepthwiseConv()
332 const Dims<4>& output_dims) { in FullyConnected()
348 float* output_data, const Dims<4>& output_dims) { in FullyConnected()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | indexed_array_analysis.h | 191 absl::Span<const int64> output_dims() const { return output_dims_; } in output_dims() function
195 std::vector<int64> output_dims, Shape shape) in ScalarIndexedArray()
226 std::vector<int64> output_dims, in ScalarIndexedConstantArray()
338 std::vector<int64> output_dims, Shape shape) { in ConstructScalarIndexedArray()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | broadcast.cc | 29 absl::Span<int64 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 | 67 std::vector<int64> output_dims(input_shape.dims()); in Compile() local
|
| /external/tensorflow/tensorflow/lite/java/src/test/native/ |
| D | interpreter_test_jni.cc | 36 TfLiteIntArray* output_dims = TfLiteIntArrayCopy(input->dims); in Java_org_tensorflow_lite_InterpreterTest_getNativeHandleForDelegate() local
|
| /external/tensorflow/tensorflow/lite/experimental/micro/kernels/ |
| D | softmax_test.cc | 33 TfLiteIntArray* output_dims = IntArrayFromInitializer(output_dims_data); in TestSoftmaxFloat() local
97 TfLiteIntArray* output_dims = IntArrayFromInitializer(output_dims_data); in TestSoftmaxQuantized() local
|
| D | depthwise_conv_test.cc | 40 TfLiteIntArray* output_dims = IntArrayFromInitializer(output_dims_data); in TestDepthwiseConvFloat() local
121 TfLiteIntArray* output_dims = IntArrayFromInitializer(output_dims_data); in TestDepthwiseConvQuantized() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | convert_trivial_transpose_to_reshape.cc | 79 std::vector<int> const& output_dims = output_array.shape().dims(); in Run() local
|
| D | reorder_reshape_transpose.cc | 173 std::vector<int> output_dims = output_shape.dims(); in Run() local
|
| D | propagate_fixed_sizes.cc | 628 std::vector<int> output_dims; in ProcessSliceOperator() local
690 auto& output_dims = *output_shape.mutable_dims(); in ProcessConcatenationOperator() local
1401 auto output_dims = output_array.mutable_shape()->mutable_dims(); in ProcessGatherOperator() local
1436 auto output_dims = output_array.mutable_shape()->mutable_dims(); in ProcessGatherNdOperator() local
1718 std::vector<int> output_dims; in ProcessSqueezeOperator() local
1796 std::vector<int>* output_dims = output_array.mutable_shape()->mutable_dims(); in ProcessTransposeOperator() local
1836 std::vector<int> output_dims; in ProcessArgMinMaxOperator() local
1959 const int output_dims = indices_dims + 1; in ProcessOneHotOperator() local
1987 std::vector<int> output_dims; in ProcessUnpackOperator() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | exp.cc | 48 TfLiteIntArray* output_dims = TfLiteIntArrayCopy(op_context.input->dims); in Prepare() local
|
| D | fake_quant.cc | 59 TfLiteIntArray* output_dims = TfLiteIntArrayCopy(op_context.input->dims); in Prepare() local
|
| D | add_n.cc | 48 TfLiteIntArray* output_dims = TfLiteIntArrayCopy(input1_dims); in Prepare() local
|
| D | where.cc | 47 TfLiteIntArray* output_dims = TfLiteIntArrayCreate(2); in ResizeOutputTensor() local
|
| D | squeeze.cc | 70 TfLiteIntArray* output_dims = in Prepare() local
|
| D | expand_dims.cc | 41 TfLiteIntArray* output_dims = TfLiteIntArrayCreate(input_dims.size + 1); in ExpandTensorDim() local
|
| D | split.cc | 62 TfLiteIntArray* output_dims = TfLiteIntArrayCopy(input->dims); in ResizeOutputTensors() local
|
| D | arg_min_max.cc | 40 TfLiteIntArray* output_dims = TfLiteIntArrayCreate(NumDimensions(input) - 1); in ResizeOutput() local
|
| D | split_v.cc | 110 TfLiteIntArray* output_dims = TfLiteIntArrayCopy(input->dims); in ResizeOutputTensors() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_softmax_op.cc | 64 memory::dims output_dims; in Compute() local
|
| D | one_hot_op.cc | 58 const int output_dims = indices_dims + 1; in Compute() local
|
| D | mkl_reshape_op.cc | 178 auto output_dims = TFShapeToMklDnnDims(shape_to); in Compute() local
|
| D | eigen_backward_spatial_convolutions.h | 419 DSizes<TensorIndex, 2> output_dims; variable
|