Home
last modified time | relevance | path

Searched defs:rhs_arg (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dbef_executable_test.cc47 auto rhs_arg = Parameter(&builder, 1, rhs_shape, "rhs"); in XLA_TEST_F() local
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
Ddeduplicate_batch_function.cc54 auto& rhs_arg = std::get<1>(arg_pair); in AreEquivalent() local
/external/tensorflow/tensorflow/core/framework/
Dfull_type_util.cc350 const FullTypeDef& rhs_arg = GetArgDefaultAny(rhs, i); in IsEqual() local
401 const FullTypeDef& rhs_arg = GetArgDefaultAny(rhs, i); in IsSubtype() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dmatrix_ops_simple_test.cc214 auto rhs_arg = Parameter(&builder, 1, rhs_shape, "rhs"); in TestImpl() local
Ddot_operation_test.cc831 auto rhs_arg = Parameter( in XLA_TYPED_TEST() local