Searched defs:first_dim_size (Results 1 – 6 of 6) sorted by relevance
77 OpInputList* data_inputs, int* first_dim_size, in CheckArgsAndAllocateResult()167 int first_dim_size; in Compute() local239 int first_dim_size; in Compute() local
53 const int32 slice_size, const int32 first_dim_size, in DynamicStitchGPUImpl()
176 const Tindex first_dim_size = static_cast<Tindex>(var.dimension(0)); in operator ()() local285 const Tindex first_dim_size = static_cast<Tindex>(var.dimension(0)); in operator ()() local612 const Tindex first_dim_size = in operator ()() local647 const Tindex first_dim_size = accum_flat.size(); in operator ()() local728 const Tindex first_dim_size = static_cast<Tindex>(var.dimension(0)); in operator ()() local1268 const Tindex first_dim_size = var.dim_size(0); in DoCompute() local1482 const Tindex first_dim_size = var.dim_size(0); in Compute() local1522 const Tindex first_dim_size = var_flat.size(); in Compute() local2488 const Tindex first_dim_size = var.dim_size(0); in Compute() local2534 const Tindex first_dim_size = var_flat.size(); in Compute() local[all …]
439 const Tindex first_dim_size = var.dimension(0); in operator ()() local500 const Tindex first_dim_size = var.dimension(0); in operator ()() local710 const Tindex first_dim_size = var.dimension(0); in operator ()() local786 const Tindex first_dim_size = var.dimension(0); in operator ()() local852 const Tindex first_dim_size = var.dimension(0); in operator ()() local
25 int64_t first_dim_size = first.sizes()[dim]; in check_cat_shape_except_dim() local
5831 int64_t first_dim_size = input_type.getDimSize(0); in matchAndRewrite() local