Home
last modified time | relevance | path

Searched refs:output_dims (Results 1 – 25 of 58) sorted by relevance

123

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h47 const Dims<4>& output_dims) { in DepthwiseConv() argument
63 bias_data, DimsToShape(output_dims), output_data); in DepthwiseConv()
73 const Dims<4>& output_dims) { in DepthwiseConv() argument
77 output_activation_max, output_data, output_dims); in DepthwiseConv()
87 const Dims<4>& output_dims) { in DepthwiseConv() argument
93 output_data, output_dims); in DepthwiseConv()
102 float* output_data, const Dims<4>& output_dims) { in DepthwiseConv() argument
105 depth_multiplier, output_data, output_dims); in DepthwiseConv()
118 const Dims<4>& output_dims) { in DepthwiseConv() argument
140 bias_data, DimsToShape(output_dims), output_data); in DepthwiseConv()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h146 const Dims<4>& output_dims) { in DepthwiseConv() argument
162 bias_data, DimsToShape(output_dims), output_data); in DepthwiseConv()
172 const Dims<4>& output_dims) { in DepthwiseConv() argument
176 output_activation_max, output_data, output_dims); in DepthwiseConv()
186 const Dims<4>& output_dims) { in DepthwiseConv() argument
192 output_data, output_dims); in DepthwiseConv()
201 float* output_data, const Dims<4>& output_dims) { in DepthwiseConv() argument
204 depth_multiplier, output_data, output_dims); in DepthwiseConv()
217 const Dims<4>& output_dims) { in DepthwiseConv() argument
239 bias_data, DimsToShape(output_dims), output_data); in DepthwiseConv()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Deigen_benchmark.h46 Dimensions output_dims(input_dims[0], // batch in SpatialConvolution()
56 static_cast<Scalar*>(device_.allocate(BufferSize(output_dims))); in SpatialConvolution()
63 Output output(output_data, output_dims); in SpatialConvolution()
82 Dimensions output_dims(input_dims[0], // batch in SpatialConvolutionBackwardInput()
94 static_cast<Scalar*>(device_.allocate(BufferSize(output_dims))); in SpatialConvolutionBackwardInput()
99 device_.memset(output_backward_data, 123, BufferSize(output_dims)); in SpatialConvolutionBackwardInput()
102 OutputBackward output_backward(output_backward_data, output_dims); in SpatialConvolutionBackwardInput()
123 Dimensions output_dims(input_dims[0], // batch in SpatialConvolutionBackwardKernel()
135 static_cast<Scalar*>(device_.allocate(BufferSize(output_dims))); in SpatialConvolutionBackwardKernel()
140 device_.memset(output_backward_data, 123, BufferSize(output_dims)); in SpatialConvolutionBackwardKernel()
[all …]
Dmkl_softmax_op.cc64 memory::dims output_dims; in Compute() local
68 output_dims = src_mkl_shape.GetSizesAsMklDnnDims(); in Compute()
71 output_dims = src_dims; in Compute()
148 output_mkl_shape.SetTfLayout(output_dims.size(), output_dims, in Compute()
153 output_tf_shape = MklDnnDimsToTFShape(output_dims); in Compute()
Deigen_backward_spatial_convolutions.h419 DSizes<TensorIndex, 2> output_dims; variable
421 output_dims[0] = kernelFilters;
422 output_dims[1] = outputRows * outputCols;
424 output_dims[1] *= out.dimension(i);
427 output_dims[1] = kernelFilters;
428 output_dims[0] = outputCols * outputRows;
430 output_dims[0] *= out.dimension(i);
442 eigen_assert(output_dims[1] == pre_contract_dims[1]);
449 eigen_assert(output_dims[0] == pre_contract_dims[0]);
480 output_backward.reshape(output_dims)
[all …]
Dconv_2d_gpu.h185 Dimension<3> output_dims;
186 output_dims[sp0] = input_dims[0];
187 output_dims[sp1] = input_dims[1];
188 output_dims[sp2] = input_dims[2];
195 Index<3> output_tensor_index = FlatToTensorIndex(output_index, output_dims);
243 Dimension<3> output_dims = {
330 TensorIndexToFlat(output_tile_origin, output_dims);
340 int output_index = output_origin_flat_index + ti * output_dims[2] + tj;
341 int output_increment = WriteRowPerPass * output_dims[2];
365 Dimension<NDIMS> output_dims,
[all …]
Dmkl_reshape_op.cc178 auto output_dims = TFShapeToMklDnnDims(shape_to); in Compute() local
179 auto output_strides = CalculateTFStrides(output_dims); in Compute()
181 output_dims, output_strides); in Compute()
190 mkl_shape_output.SetTfLayout(output_dims.size(), output_dims, in Compute()
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Dbroadcast.cc29 absl::Span<int64 const> output_dims) { in BroadcastTo() argument
35 if (input_dims == output_dims) { in BroadcastTo()
39 if (input_dims.size() > output_dims.size()) { in BroadcastTo()
43 absl::StrJoin(output_dims, ","), "]"); in BroadcastTo()
49 for (auto output_it = output_dims.rbegin(); output_it != output_dims.rend(); in BroadcastTo()
56 " to [", absl::StrJoin(output_dims, ","), in BroadcastTo()
85 if (broadcast_shape != output_dims) { in BroadcastTo()
86 output = xla::Reshape(output, output_dims); in BroadcastTo()
/external/tensorflow/tensorflow/compiler/xla/service/
Dindexed_array_analysis.cc74 StrJoin(indexed_array->output_dims(), ","), "])"); in ToString()
189 absl::Span<const int64> output_dims, Shape shape) { in FoldGatherOfGather() argument
208 for (int64 gather_dim : source->output_dims()) { in FoldGatherOfGather()
215 for (int64 output_dim : output_dims) { in FoldGatherOfGather()
221 FindIndex(source->output_dims(), source_dim); in FoldGatherOfGather()
222 CHECK_NE(source_dim_for_index_array, source->output_dims().size()); in FoldGatherOfGather()
291 std::vector<int64> output_dims; in ComputeArrayForGather() local
294 output_dims.push_back(i); in ComputeArrayForGather()
299 if (absl::c_linear_search(indexed->output_dims(), source_dim)) { in ComputeArrayForGather()
300 return FoldGatherOfGather(indexed, indices, source_dim, output_dims, in ComputeArrayForGather()
[all …]
Dindexed_array_analysis.h191 absl::Span<const int64> output_dims() const { return output_dims_; } in output_dims() function
195 std::vector<int64> output_dims, Shape shape) in ScalarIndexedArray() argument
199 output_dims_(std::move(output_dims)), in ScalarIndexedArray()
226 std::vector<int64> output_dims, in ScalarIndexedConstantArray() argument
229 std::move(output_dims), std::move(shape)) { in ScalarIndexedConstantArray()
306 absl::Span<const int64> output_dims, Shape shape);
338 std::vector<int64> output_dims, Shape shape) { in ConstructScalarIndexedArray() argument
341 std::move(output_dims), in ConstructScalarIndexedArray()
345 std::move(output_dims), in ConstructScalarIndexedArray()
/external/tensorflow/tensorflow/lite/kernels/
Dexpand_dims.cc41 TfLiteIntArray* output_dims = TfLiteIntArrayCreate(input_dims.size + 1); in ExpandTensorDim() local
42 for (int i = 0; i < output_dims->size; ++i) { in ExpandTensorDim()
44 output_dims->data[i] = input_dims.data[i]; in ExpandTensorDim()
46 output_dims->data[i] = 1; in ExpandTensorDim()
48 output_dims->data[i] = input_dims.data[i - 1]; in ExpandTensorDim()
52 return context->ResizeTensor(context, output, output_dims); in ExpandTensorDim()
Dwhere.cc47 TfLiteIntArray* output_dims = TfLiteIntArrayCreate(2); in ResizeOutputTensor() local
48 output_dims->data[0] = true_count; in ResizeOutputTensor()
49 output_dims->data[1] = cond_rank; in ResizeOutputTensor()
50 return context->ResizeTensor(context, output_tensor, output_dims); in ResizeOutputTensor()
Done_hot.cc47 output_dims = indices_dims + 1; in OneHotContext()
57 int output_dims; member
107 TfLiteIntArray* output_size = TfLiteIntArrayCreate(op_context.output_dims); in ResizeOutputTensor()
108 for (int i = 0; i < op_context.output_dims; ++i) { in ResizeOutputTensor()
143 op_context.axis < op_context.output_dims); in Prepare()
Dsqueeze.cc70 TfLiteIntArray* output_dims = in Prepare() local
74 output_dims->data[out_idx++] = input_dims->data[in_idx]; in Prepare()
77 return context->ResizeTensor(context, op_context.output, output_dims); in Prepare()
Dsplit.cc62 TfLiteIntArray* output_dims = TfLiteIntArrayCopy(input->dims); in ResizeOutputTensors() local
63 output_dims->data[axis_value] = slice_size; in ResizeOutputTensors()
65 TF_LITE_ENSURE_STATUS(context->ResizeTensor(context, output, output_dims)); in ResizeOutputTensors()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtile_ops.cc67 std::vector<int64> output_dims(input_shape.dims()); in Compile() local
71 "] >= 0, but got ", output_dims[i])); in Compile()
72 output_dims[i] = input_shape.dim_size(i) * multiples[i]; in Compile()
96 XlaHelpers::Zero(ctx->builder(), ctx->input_type(0)), output_dims); in Compile()
105 auto result = BroadcastTo(ctx->Input("input"), output_dims); in Compile()
Done_hot_op.cc39 const int output_dims = indices_dims + 1; in Compile() local
43 ctx, axis_ == -1 || (axis_ >= 0 && axis_ < output_dims), in Compile()
45 output_dims, "). But received: ", axis_)); in Compile()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconv_depthwise_test.cc41 std::vector<int64> output_dims; member
76 config.output_dims = {batch, activation_size + kernel_size - 1, in GetConv2DTestCases()
83 config.output_dims = {batch, feature / 32, in GetConv2DTestCases()
87 config.output_dims = {batch, activation_size - kernel_size + 1, in GetConv2DTestCases()
121 absl::StrJoin(spec.output_dims, "x"), "_output_layout_", in DepthwiseConvolution2DTestDataToString()
153 absl::StrJoin(spec.output_dims, ","), in BuildHloTextDepthwiseConvolution2D()
178 absl::StrJoin(spec.output_dims, ","), in BuildHloTextDepthwiseConvolution2D()
202 absl::StrJoin(spec.output_dims, ","), in BuildHloTextDepthwiseConvolution2D()
Dgrouped_convolution_test.cc42 std::vector<int64> output_dims; member
85 config.output_dims = {batch, activation_size + kernel_size - 1, in GetConv2DTestCases()
92 config.output_dims = {batch, output_feature / 32, in GetConv2DTestCases()
96 config.output_dims = {batch, activation_size - kernel_size + 1, in GetConv2DTestCases()
130 absl::StrJoin(spec.output_dims, "x"), "_output_layout_", in GroupedConvolution2DTestDataToString()
163 absl::StrJoin(spec.output_dims, ","), in BuildHloTextGroupedConvolution2D()
189 absl::StrJoin(spec.output_dims, ","), in BuildHloTextGroupedConvolution2D()
214 absl::StrJoin(spec.output_dims, ","), in BuildHloTextGroupedConvolution2D()
Dconv_depthwise_backprop_filter_test.cc39 std::vector<int64> output_dims; member
72 config.output_dims = {output_space_size, output_space_size, feature, 1}; in GetConv2DTestCases()
95 config.output_dims = {output_space_size, output_space_size, feature, 1}; in GetConv2DTestCases()
118 absl::StrJoin(spec.output_dims, "x"), data_type, "_output_layout_", in BatchGroupedConvolution2DTestDataToString()
145 absl::StrJoin(spec.output_dims, ","), in BuildHloTextBatchGroupedConvolution2D()
/external/tensorflow/tensorflow/contrib/grid_rnn/python/ops/
Dgrid_rnn_cell.py54 output_dims=None, argument
120 self._config = _parse_rnn_config(num_dims, input_dims, output_dims,
335 output_dims=0,
363 output_dims=0,
387 output_dims=0,
418 output_dims=0,
450 output_dims=0,
483 output_dims=0,
520 output_dims=0,
549 output_dims=0,
[all …]
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/
Dsoftmax_test.cc33 TfLiteIntArray* output_dims = IntArrayFromInitializer(output_dims_data); in TestSoftmaxFloat() local
34 const int output_dims_count = ElementCount(*output_dims); in TestSoftmaxFloat()
41 CreateFloatTensor(output_data, output_dims, "output_tensor"), in TestSoftmaxFloat()
97 TfLiteIntArray* output_dims = IntArrayFromInitializer(output_dims_data); in TestSoftmaxQuantized() local
98 const int output_dims_count = ElementCount(*output_dims); in TestSoftmaxQuantized()
106 CreateQuantizedTensor(output_data, output_dims, "output_tensor", in TestSoftmaxQuantized()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dconvert_trivial_transpose_to_reshape.cc79 std::vector<int> const& output_dims = output_array.shape().dims(); in Run() local
97 1, static_cast<int>(output_dims.size())}; in Run()
101 shape_buffer.data = output_dims; in Run()
Dpropagate_fixed_sizes.cc628 std::vector<int> output_dims; in ProcessSliceOperator() local
634 output_dims.push_back(size); in ProcessSliceOperator()
637 *output_array.mutable_shape()->mutable_dims() = output_dims; in ProcessSliceOperator()
690 auto& output_dims = *output_shape.mutable_dims(); in ProcessConcatenationOperator() local
692 output_dims[op->axis] = concat_size; in ProcessConcatenationOperator()
1401 auto output_dims = output_array.mutable_shape()->mutable_dims(); in ProcessGatherOperator() local
1403 output_dims->push_back(input_shape.dims(dim)); in ProcessGatherOperator()
1406 output_dims->push_back(indices_shape.dims(dim)); in ProcessGatherOperator()
1409 output_dims->push_back(input_shape.dims(dim)); in ProcessGatherOperator()
1436 auto output_dims = output_array.mutable_shape()->mutable_dims(); in ProcessGatherNdOperator() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmatrix.cc231 std::vector<int64> output_dims; in Einsum() local
232 output_dims.reserve(output_rank); in Einsum()
234 output_dims.push_back(output_dimension_number(x_config[d])); in Einsum()
237 output_dims.push_back(output_dimension_number(x_config[d])); in Einsum()
240 output_dims.push_back(output_dimension_number(y_config[d])); in Einsum()
245 transpose_dims[output_dims[i]] = i; in Einsum()

123