HomeSort by relevance Sort by last modified time
    Searched defs:out_dim (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/core/framework/
shape_inference_testutil.cc 157 DimensionHandle out_dim = c.Dim(out, j); local
163 if (c.Dim(in, j).SameHandle(out_dim)) {
177 } else if (c.ValueKnown(out_dim)) {
179 c.Value(out_dim), err_suffix);
220 } else if (value != c.Value(out_dim)) {
222 c.DebugString(out_dim), err_suffix);
  /external/tensorflow/tensorflow/python/framework/
cpp_shape_inference.cc 39 auto* out_dim = out->add_dim(); local
41 out_dim->set_size(c->Value(d));
43 out_dim->set_size(-1);
  /external/tensorflow/tensorflow/core/kernels/
ragged_gather_op.cc 166 int out_dim = dim + indices_in.dims() - 1; local
167 if (out_dim >= 0) {
168 int64 delta = out_splits->at(out_dim).back() - splits(start);
170 out_splits->at(out_dim).push_back(splits(j + 1) + delta);
eigen_spatial_convolutions_test.cc 1307 DSizes<ptrdiff_t, 3> out_dim; local
    [all...]

Completed in 176 milliseconds