Searched defs:first_dim_size (Results 1 – 3 of 3) sorted by relevance
77 OpInputList* data_inputs, int* first_dim_size, in CheckArgsAndAllocateResult()166 int first_dim_size; in Compute() local238 int first_dim_size; in Compute() local
53 const int32 slice_size, const int32 first_dim_size, in DynamicStitchGPUImpl()
842 const Tindex first_dim_size = var.dim_size(0); in DoCompute() local1037 const Tindex first_dim_size = var.dim_size(0); in Compute() local1077 const Tindex first_dim_size = var_flat.size(); in Compute() local1386 const Tindex first_dim_size = var.dim_size(0); in Compute() local1415 const Tindex first_dim_size = accum_flat.size(); in Compute() local1544 const Tindex first_dim_size = var.dim_size(0); in Compute() local1587 const Tindex first_dim_size = accum_flat.size(); in Compute() local1840 const Tindex first_dim_size = var.dim_size(0); in Compute() local1886 const Tindex first_dim_size = var_flat.size(); in Compute() local2203 const Tindex first_dim_size = var.dim_size(0); in Compute() local[all …]