Searched defs:dim_name (Results 1 – 6 of 6) sorted by relevance
73 const auto dim_name = params_layout->sharding_spec(i); in ExpandOp() local89 auto dim_name = params_layout->sharding_spec(i); in ExpandOp() local
159 const std::string& dim_name(int64 index) const { in dim_name() function
796 const std::string& dim_name) { in IndexOfMeshDimension()1088 std::string dim_name = dim.sharding_spec(); in ToString() local
447 auto dim_name = x_config[i]; in Einsum() local478 auto dim_name = y_config[i]; in Einsum() local
886 static const char *dim_name[] = { in print_intrinsic_instr() local
410 llvm::StringRef crop_b_name) -> LogicalResult { in verify()