Searched refs:base_inner_size (Results 1 – 5 of 5) sorted by relevance
55 int64_t base_inner_size = 1; in Concatenation() local57 base_inner_size *= output_shape.Dims(i); in Concatenation()63 const int copy_size = input_shapes[i]->Dims(axis) * base_inner_size; in Concatenation()106 int64_t base_inner_size = 1; in ConcatenationWithScaling() local108 base_inner_size *= output_shape.Dims(i); in ConcatenationWithScaling()115 const int copy_size = input_shapes[i]->Dims(axis) * base_inner_size; in ConcatenationWithScaling()
859 int64_t base_inner_size = 1; in Split() local861 base_inner_size *= input_shape.Dims(i); in Split()867 const int copy_size = output_shapes[i]->Dims(axis) * base_inner_size; in Split()
50 int64_t base_inner_size = 1; in SplitImpl() local52 base_inner_size *= input_dims->data[i]; in SplitImpl()60 const int copy_size = output_dims->data[axis] * base_inner_size; in SplitImpl()
52 int64_t base_inner_size = 1; in SplitImpl() local54 base_inner_size *= input_dims->data[i]; in SplitImpl()64 output_tensor->dims->data[axis_value] * base_inner_size; in SplitImpl()
709 const int64_t base_inner_size = in ConstFoldConcatenateOpDense() local721 const int64_t inner_size = dim_size * base_inner_size; in ConstFoldConcatenateOpDense()