Home
last modified time | relevance | path

Searched defs:token_shape (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_invariant_code_motion_test.cc254 auto token_shape = ShapeUtil::MakeTokenShape(); in TEST_F() local
303 auto token_shape = ShapeUtil::MakeTokenShape(); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconcat_test.cc429 auto token_shape = ShapeUtil::MakeTokenShape(); in XLA_TEST_F() local
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc1170 xla::Shape token_shape = xla::TypeToShape(result_types[num_results - 1]); in ExportXlaOp() local
1283 xla::Shape token_shape = xla::TypeToShape(result_types[num_results - 1]); in ExportXlaOp() local