Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/tests/
H A Dxla_compilation_cache_test_helper.cc30 Tensor CreateInputTensor(const TensorShape& shape, float offset) { in CreateInputTensor() function
95 Tensor input_a = CreateInputTensor(shape, 0); in ExecuteWithBatch()
96 Tensor input_b = CreateInputTensor(shape, shape.num_elements()); in ExecuteWithBatch()
97 Tensor input_c = CreateInputTensor(shape, 2 * shape.num_elements()); in ExecuteWithBatch()
118 Tensor input_a = CreateInputTensor(shape, 0); in ExecuteWithBatch()
119 Tensor input_b = CreateInputTensor(shape, shape.num_elements()); in ExecuteWithBatch()
120 Tensor input_c = CreateInputTensor(shape, 2 * shape.num_elements()); in ExecuteWithBatch()