Searched defs:last_dim (Results 1 – 9 of 9) sorted by relevance
42 int last_dim = input_shape.dims() - 1; in Compile() local
178 int last_dim = dims.size() - 1; in Compile() local
28 int last_dim = input_shape.dimensions_size() - 1; in TopK() local
52 const auto last_dim = input.shape().dims() - 1; in Compute() local
95 const int last_dim = input_tensor.dim_size(input_tensor.dims() - 1); in operator ()() local
113 DimensionHandle last_dim = c->Dim(input, 3); in __anon4650176f0302() local147 DimensionHandle last_dim = c->Dim(input, 3); in __anon4650176f0402() local1216 DimensionHandle last_dim = c->Dim(input, -1); in TopKShapeFn() local1272 DimensionHandle last_dim = c->Dim(input, -1); in __anon4650176f2602() local1532 DimensionHandle last_dim = c->Dim(input, 3); in __anon4650176f3002() local
102 DimensionHandle last_dim; in ColorspaceShapeFn() local
2658 DimensionHandle last_dim = c->Dim(new_shape, -1); in __anondb9326b23b02() local3124 ShapeHandle last_dim = c->Vector(c->Dim(inputs, -1)); in __anondb9326b24a02() local
4172 def _unknown_dim_invalid_input(self, last_dim): argument