Home
last modified time | relevance | path

Searched refs:PartitionLmhloOperandsAndOutputs (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emission_utils_test.cc43 EXPECT_EQ(2, PartitionLmhloOperandsAndOutputs(op)); in TEST()
60 EXPECT_EQ(2, PartitionLmhloOperandsAndOutputs(op)); in TEST()
77 EXPECT_EQ(2, PartitionLmhloOperandsAndOutputs(op)); in TEST()
Dir_emission_utils.h252 int PartitionLmhloOperandsAndOutputs(mlir::Operation* op);
Dir_emission_utils.cc657 int PartitionLmhloOperandsAndOutputs(mlir::Operation* op) { in PartitionLmhloOperandsAndOutputs() function
677 int output_start = PartitionLmhloOperandsAndOutputs(op); in GetHloOperands()
697 int output_start = PartitionLmhloOperandsAndOutputs(op); in GetHloOutputs()