Searched refs:dim_name (Results 1 – 3 of 3) sorted by relevance
414 auto dim_name = x_config[i]; in Einsum() local415 const int64 rhs_dim = rhs_dimension_number(dim_name); in Einsum()417 if (is_batch_dim(dim_name)) { in Einsum()428 } else if (is_contracting(dim_name)) { in Einsum()445 auto dim_name = y_config[i]; in Einsum() local446 if (!is_batch_dim(dim_name) && !is_contracting(dim_name)) { in Einsum()
360 llvm::StringRef dim_name, in Verify()372 << "requires output " << dim_name << " (dimension " in Verify()374 << dim_name << " (dimension " << spatial_dim_index in Verify()375 << ") * block_size, but got output " << dim_name << " " in Verify()376 << output_dim << ", input " << dim_name << " " << input_dim in Verify()385 << "requires output " << dim_name << " (dimension " in Verify()386 << spatial_dim_index << ") to be equal to input " << dim_name in Verify()389 << dim_name << " " << output_dim << ", input " << dim_name in Verify()
886 static const char *dim_name[] = { in print_intrinsic_instr() local898 assert(dim < ARRAY_SIZE(dim_name) && dim_name[dim]); in print_intrinsic_instr()899 fprintf(fp, " image_dim=%s", dim_name[dim]); in print_intrinsic_instr()