Searched defs:last_dim (Results 1 – 14 of 14) sorted by relevance
35 int last_dim = input_shape.dims() - 1; in Compile() local
32 int last_dim = input_shape.dimensions_size() - 1; in TopK() local157 int last_dim = input_shape.dimensions_size() - 1; in TopKWithPartitions() local
43 TF_DimensionHandle* last_dim = TF_NewDimensionHandle(); in ComputeNewShape() local
73 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
241 const auto last_dim = input.shape().dims() - 1; in Compute() local
114 DimensionHandle last_dim = c->Dim(input, 3); in __anon1bbab8940202() local148 DimensionHandle last_dim = c->Dim(input, 3); in __anon1bbab8940302() local1344 DimensionHandle last_dim = c->Dim(input, -1); in TopKShapeFn() local1495 DimensionHandle last_dim = c->Dim(input, -1); in __anon1bbab8942102() local1737 DimensionHandle last_dim = c->Dim(input, 3); in __anon1bbab8942902() local
136 DimensionHandle last_dim; in ColorspaceShapeFn() local
3343 ShapeHandle last_dim = c->Vector(c->Dim(inputs, -1)); in __anon091a71b74b02() local
135 unsigned last_dim = num_loops - 1; in runOnOperation() local
992 int64_t last_dim = -1; in matchAndRewrite() local994 auto is_last_dim_compatible = [](const Value &v, int64_t &last_dim) { in matchAndRewrite()
2104 int last_dim = (*dims)[dims_size - 1]; in ProcessMatrixDiagOperator() local
3453 const int last_dim = in Softmax() local3597 const int last_dim = in SoftmaxInt8LUT() local3807 const int last_dim = in LogSoftmax() local
3677 int last_dim = std::max(static_cast<int>(input_shape.size()) - 1, 0); in rewriteWithUnknownBegin() local