Searched refs:branch_index_arg (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | conditional_test.cc | 198 auto branch_index_arg = CreateR0Parameter<int32>(bi, 0, "branch_index_arg", in XLA_TEST_P() local
216 ComputeAndCompareR0<float>(&builder, expected, {branch_index_arg.get()}, in XLA_TEST_P()
241 auto branch_index_arg = CreateR0Parameter<int32>(bi, 0, "branch_index_arg", in XLA_TEST_P() local
268 ComputeAndCompareR0<float>(&builder, expected, {branch_index_arg.get()}, in XLA_TEST_P()
416 auto branch_index_arg = in XLA_TEST_P() local
442 {branch_index_arg.get()}, error_spec_); in XLA_TEST_P()
|