Searched refs:split_dim_index (Results 1 – 1 of 1) sorted by relevance
457 const int split_dim_index = std::distance( in EmitHaloExchange() local460 if (input_tensor_shape[split_dim_index] < halo_size) in EmitHaloExchange()475 halo_exchange_tensor_shape.push_back(index == split_dim_index ? halo_size in EmitHaloExchange()517 begin_left[split_dim_index] = input_tensor_shape[split_dim_index] - halo_size; in EmitHaloExchange()523 size[split_dim_index] = halo_size; in EmitHaloExchange()578 final_shape[split_dim_index] = final_shape[split_dim_index] + 2 * halo_size; in EmitHaloExchange()583 CreateIntScalarConst(split_dim_index, builder, location); in EmitHaloExchange()