OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:dim_i
(Results
1 - 4
of
4
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
cudnn_pooling_gpu.cc
82
const auto
dim_i
= static_cast<se::dnn::DimIndex>(i);
local
83
pooling_desc.set_window(
dim_i
, window[i]);
84
pooling_desc.set_stride(
dim_i
, stride[i]);
85
pooling_desc.set_padding(
dim_i
, padding[i]);
86
input_desc.set_spatial_dim(
dim_i
,
88
output_desc.set_spatial_dim(
dim_i
,
203
const auto
dim_i
= static_cast<se::dnn::DimIndex>(i);
local
204
pooling_desc.set_window(
dim_i
, window[i]);
205
pooling_desc.set_stride(
dim_i
, stride[i]);
206
pooling_desc.set_padding(
dim_i
, padding[i])
[
all
...]
/external/tensorflow/tensorflow/core/util/
strided_slice_op.cc
260
int64
dim_i
= input_shape.dim_size(i);
local
265
if (
dim_i
== -1) {
273
{stride_i > 0 ? 0 : -1, stride_i > 0 ?
dim_i
:
dim_i
- 1}};
275
auto canonical = [stride_i,
dim_i
, masks, valid_range](int64 x, int c) {
279
int64 x_fwd = x < 0 ?
dim_i
+ x : x; // make negative indices positive
299
int64 x_fwd = begin_i < 0 ?
dim_i
+ begin_i : begin_i;
302
if (x_fwd < 0 || x_fwd >=
dim_i
) {
312
stride_i == 1 && begin_i == 0 && end_i ==
dim_i
;
334
if (
dim_i
>= 0)
[
all
...]
tensor_format_test.cc
60
int i() const { return
dim_i
; }
64
int dim_h, dim_w,
dim_i
, dim_o;
member in struct:tensorflow::FilterDimMap
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
convert_nodes.cc
[
all
...]
Completed in 567 milliseconds