Home
last modified time | relevance | path

Searched defs:last_dim (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtopk_op.cc42 int last_dim = input_shape.dims() - 1; in Compile() local
Ddiag_op.cc178 int last_dim = dims.size() - 1; in Compile() local
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dsorting.cc28 int last_dim = input_shape.dimensions_size() - 1; in TopK() local
/external/tensorflow/tensorflow/core/kernels/
Dquantized_bias_add_op.cc52 const auto last_dim = input.shape().dims() - 1; in Compute() local
Dnth_element_op.cc95 const int last_dim = input_tensor.dim_size(input_tensor.dims() - 1); in operator ()() local
/external/tensorflow/tensorflow/core/ops/
Dnn_ops.cc113 DimensionHandle last_dim = c->Dim(input, 3); in __anon3e672dd80302() local
147 DimensionHandle last_dim = c->Dim(input, 3); in __anon3e672dd80402() local
1216 DimensionHandle last_dim = c->Dim(input, -1); in TopKShapeFn() local
1272 DimensionHandle last_dim = c->Dim(input, -1); in __anon3e672dd82602() local
1532 DimensionHandle last_dim = c->Dim(input, 3); in __anon3e672dd83002() local
Dimage_ops.cc102 DimensionHandle last_dim; in ColorspaceShapeFn() local
Darray_ops.cc2658 DimensionHandle last_dim = c->Dim(new_shape, -1); in __anon7c94107b3b02() local
3124 ShapeHandle last_dim = c->Vector(c->Dim(inputs, -1)); in __anon7c94107b4a02() local
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dlayers_test.py4172 def _unknown_dim_invalid_input(self, last_dim): argument