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 __anon3e672dd80302() local147 DimensionHandle last_dim = c->Dim(input, 3); in __anon3e672dd80402() local1216 DimensionHandle last_dim = c->Dim(input, -1); in TopKShapeFn() local1272 DimensionHandle last_dim = c->Dim(input, -1); in __anon3e672dd82602() local1532 DimensionHandle last_dim = c->Dim(input, 3); in __anon3e672dd83002() local
102 DimensionHandle last_dim; in ColorspaceShapeFn() local
2658 DimensionHandle last_dim = c->Dim(new_shape, -1); in __anon7c94107b3b02() local3124 ShapeHandle last_dim = c->Vector(c->Dim(inputs, -1)); in __anon7c94107b4a02() local
4172 def _unknown_dim_invalid_input(self, last_dim): argument