Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmatrix_test.cc166 EXPECT_TRUE(ValidateEinsumNumericDimensions( in XLA_TEST_F()
184 EXPECT_FALSE(ValidateEinsumNumericDimensions( in XLA_TEST_F()
Dmatrix.h92 Status ValidateEinsumNumericDimensions(absl::Span<const int64> x_config,
Dmatrix.cc118 Status ValidateEinsumNumericDimensions(absl::Span<const int64> x_config, in ValidateEinsumNumericDimensions() function
166 ValidateEinsumNumericDimensions(x_config, y_config, output_config)); in Einsum()