Searched refs:arg_param (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convert_test.cc | 214 auto arg_param = Parameter(&builder, 0, arg_literal.shape(), "arg_param"); in XLA_TEST_F() local
218 ConvertElementType(arg_param, F32); in XLA_TEST_F()
233 auto arg_param = Parameter(&builder, 0, arg_literal.shape(), "arg_param"); in XLA_TEST_F() local
237 ConvertElementType(arg_param, F32); in XLA_TEST_F()
251 auto arg_param = Parameter(&builder, 0, arg_literal.shape(), "arg_param"); in XLA_TEST_F() local
255 ConvertElementType(arg_param, U32); in XLA_TEST_F()
268 auto arg_param = Parameter(&builder, 0, arg_literal.shape(), "arg_param"); in XLA_TEST_F() local
272 ConvertElementType(arg_param, S64); in XLA_TEST_F()
285 auto arg_param = Parameter(&builder, 0, arg_literal.shape(), "arg_param"); in XLA_TEST_F() local
289 ConvertElementType(arg_param, S64); in XLA_TEST_F()
[all …]
|