Searched refs:input_shape_t (Results 1 – 2 of 2) sorted by relevance
375 auto input_shape_t = input_shape->vec<int64>(); in Compute() local376 const int64 N = input_shape_t(0); in Compute()386 for (int d = 1; d < rank; d++) output_shape_t(d - 1) = input_shape_t(d); in Compute()
265 auto input_shape_t = input_shape->vec<int64>(); in Compute() local266 const int64 N = input_shape_t(0); in Compute()277 input_shape_t.data() + 1, in Compute()