OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:post_contract_dims
(Results
1 - 5
of
5
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
runtime_conv2d_impl.h
62
Eigen::DSizes<int64, 4>
post_contract_dims
;
local
63
post_contract_dims
[0] = input_batch;
64
post_contract_dims
[1] = output_rows;
65
post_contract_dims
[2] = output_cols;
66
post_contract_dims
[3] = kernel_filters;
82
.reshape(
post_contract_dims
);
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
eigen_spatial_convolutions.h
190
DSizes<TensorIndex, NumDims>
post_contract_dims
;
local
192
post_contract_dims
[0] = kernelFilters;
193
post_contract_dims
[1] = out_height;
194
post_contract_dims
[2] = out_width;
196
post_contract_dims
[i] = in.dimension(i);
199
post_contract_dims
[NumDims - 1] = kernelFilters;
200
post_contract_dims
[NumDims - 2] = out_height;
201
post_contract_dims
[NumDims - 3] = out_width;
203
post_contract_dims
[i] = in.dimension(i);
222
.reshape(
post_contract_dims
);
[
all
...]
/external/tensorflow/tensorflow/core/kernels/
eigen_cuboid_convolution.h
187
DSizes<TensorIndex, NumDims>
post_contract_dims
;
local
189
post_contract_dims
[0] = kernelFilters;
190
post_contract_dims
[1] = out_depth;
191
post_contract_dims
[2] = out_height;
192
post_contract_dims
[3] = out_width;
194
post_contract_dims
[i] = in.dimension(i);
197
post_contract_dims
[NumDims - 1] = kernelFilters;
198
post_contract_dims
[NumDims - 2] = out_depth;
199
post_contract_dims
[NumDims - 3] = out_height;
200
post_contract_dims
[NumDims - 4] = out_width
[
all
...]
eigen_backward_spatial_convolutions.h
249
DSizes<TensorIndex, NumDims>
post_contract_dims
;
local
251
post_contract_dims
[0] = kernelChannels;
252
post_contract_dims
[1] = inputRows;
253
post_contract_dims
[2] = inputCols;
255
post_contract_dims
[i] = out.dimension(i);
258
post_contract_dims
[NumDims - 1] = kernelChannels;
259
post_contract_dims
[NumDims - 2] = inputRows;
260
post_contract_dims
[NumDims - 3] = inputCols;
262
post_contract_dims
[i] = out.dimension(i);
280
.reshape(
post_contract_dims
),
[
all
...]
eigen_backward_cuboid_convolutions.h
245
DSizes<TensorIndex, NumDims>
post_contract_dims
;
local
247
post_contract_dims
[0] = kernelChannels;
248
post_contract_dims
[1] = inputPlanes;
249
post_contract_dims
[2] = inputRows;
250
post_contract_dims
[3] = inputCols;
252
post_contract_dims
[i] = out.dimension(i);
255
post_contract_dims
[NumDims - 1] = kernelChannels;
256
post_contract_dims
[NumDims - 2] = inputPlanes;
257
post_contract_dims
[NumDims - 3] = inputRows;
258
post_contract_dims
[NumDims - 4] = inputCols
[
all
...]
Completed in 260 milliseconds