Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/keras/saving/
Dsaved_model_test.py330 input_arr = np.random.random((1, 3))
340 model.train_on_batch(input_arr, target_arr)
342 ref_loss, ref_mae = model.evaluate(input_arr, target_arr)
344 ref_predict = model.predict(input_arr)
359 {inputs[input_name]: input_arr})
377 inputs[input_name]: input_arr,
405 [outputs, train_op], {inputs[input_name]: input_arr,
439 input_arr = np.random.random((1, 3))
453 model.train_on_batch(input_arr, target_arr)
461 clone.train_on_batch(input_arr, target_arr)
[all …]
/external/tensorflow/tensorflow/compiler/xla/tests/
Dexhaustive_op_test.cc260 absl::Span<T> input_arr = input_literal.data<T>(); in RunImpl() local
267 input_arr[i] = T{0}; in RunImpl()
269 input_arr[i] = absl::bit_cast<T>(input_val); in RunImpl()
342 absl::Span<const T> input_arr = input_literal.data<T>(); in ExpectNear() local
344 ASSERT_EQ(result_arr.size(), input_arr.size()); in ExpectNear()
350 for (int64 i = 0; i < input_arr.size(); ++i) { in ExpectNear()
351 T input = input_arr[i]; in ExpectNear()
Dreduce_test.cc896 std::vector<T> input_arr(num_elems, std::numeric_limits<T>::lowest()); in DoTest() local
897 auto input_literal = LiteralUtil::CreateR1<T>(input_arr); in DoTest()