Searched refs:update_input_tensor (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | sharding_util_ops_test.cc | 593 Tensor update_input_tensor(update_data_type, update_input_shape); in TEST() local
594 test::FillIota<float>(&update_input_tensor, /*val=*/0.f); in TEST()
595 Node* update_input = test::graph::Constant(&graph, update_input_tensor); in TEST()
640 Tensor update_input_tensor(update_data_type, update_input_shape); in TEST() local
641 test::FillIota<float>(&update_input_tensor, /*val=*/0.f); in TEST()
642 Node* update_input = test::graph::Constant(&graph, update_input_tensor); in TEST()
676 Tensor update_input_tensor(update_data_type, update_input_shape); in TEST() local
677 test::FillIota<float>(&update_input_tensor, /*val=*/0.f); in TEST()
678 Node* update_input = test::graph::Constant(&graph, update_input_tensor); in TEST()
711 Tensor update_input_tensor(update_data_type, update_input_shape); in TEST() local
[all …]
|