Home
last modified time | relevance | path

Searched refs:init_input_tensor (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dsharding_util_ops_test.cc627 Tensor init_input_tensor(init_data_type, init_input_shape); in TEST() local
628 test::FillIota<int32>(&init_input_tensor, /*val=*/0); in TEST()
629 Node* input = test::graph::Constant(&graph, init_input_tensor); in TEST()
746 Tensor init_input_tensor(data_type, init_input_shape); in TEST() local
747 test::FillFn<float>(&init_input_tensor, [](int unused) { return -1.f; }); in TEST()
748 Node* init_input = test::graph::Constant(&graph, init_input_tensor); in TEST()
833 Tensor init_input_tensor(data_type, input_shapes.front()); in CreateConcatResourceGraph() local
834 test::FillFn<int32>(&init_input_tensor, [](int unused) { return -1; }); in CreateConcatResourceGraph()
835 Node* init_input = test::graph::Constant(graph, init_input_tensor); in CreateConcatResourceGraph()