OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:indices_dims
(Results
1 - 4
of
4
) sorted by null
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
one_hot_op.cc
38
const int
indices_dims
= indices_shape.dims();
variable
39
const int output_dims =
indices_dims
+ 1;
56
const int axis = (axis_ == -1) ?
indices_dims
: axis_;
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
scatter.cc
47
gtl::ArraySlice<int64>
indices_dims
=
local
56
TF_RET_CHECK(!
indices_dims
.empty());
57
num_index_dims =
indices_dims
.back();
65
indices_dims
.pop_back();
69
for (int64 dim :
indices_dims
) {
/external/tensorflow/tensorflow/core/kernels/
one_hot_op.cc
57
const int
indices_dims
= indices_shape.dims();
variable
58
const int output_dims =
indices_dims
+ 1;
75
const int axis = (axis_ == -1) ?
indices_dims
: axis_;
/external/tensorflow/tensorflow/compiler/tf2xla/
xla_helpers.cc
233
const int
indices_dims
= indices_shape.dims();
local
234
const int output_dims =
indices_dims
+ 1;
Completed in 266 milliseconds