Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/keras/saving/saved_model/
Dsaved_model_test.py160 input_arr = constant_op.constant(
162 self.assertAllClose(self.evaluate(model(input_arr)),
163 self.evaluate(loaded(input_arr)))
173 sorted(self.evaluate(model.get_losses_for(input_arr))),
174 sorted(self.evaluate(loaded.get_losses_for(input_arr))))
217 input_arr = np.random.random((1, 3))
222 model.train_on_batch(input_arr, target_arr)
235 model(input_arr) # Calls model eagerly, creating eager losses.
246 input_arr = array_ops.ones((4, 3))
250 self.assertAllEqual(input_arr, loaded(input_arr))
[all …]
Drevive_test.py213 input_arr = constant_op.constant(
217 input_arr = sparse_ops.from_dense(input_arr)
219 self.assertAllClose(model(input_arr), revived(input_arr))
/external/tensorflow/tensorflow/python/keras/saving/
Dsaved_model_experimental_test.py339 input_arr = np.random.random((1, 3))
349 model.train_on_batch(input_arr, target_arr)
351 ref_loss, ref_mae = model.evaluate(input_arr, target_arr)
353 ref_predict = model.predict(input_arr)
368 {inputs[input_name]: input_arr})
386 inputs[input_name]: input_arr,
414 [outputs, train_op], {inputs[input_name]: input_arr,
448 input_arr = np.random.random((1, 3))
462 model.train_on_batch(input_arr, target_arr)
470 clone.train_on_batch(input_arr, target_arr)
[all …]
/external/tensorflow/tensorflow/compiler/xla/tests/
Ddot_operation_test.cc1686 Array3D<float> input_arr(2, 3, 2); in XLA_TEST_F() local
1688 input_arr.FillIota(0); in XLA_TEST_F()
1693 AddParam(LiteralUtil::CreateR3FromArray3D<float>(input_arr), &builder); in XLA_TEST_F()
1703 Array3D<float> input_arr(2, 3, 2); in XLA_TEST_F() local
1705 input_arr.FillIota(0); in XLA_TEST_F()
1710 AddParam(LiteralUtil::CreateR3FromArray3D<float>(input_arr), &builder); in XLA_TEST_F()
1724 Array4D<float> input_arr(2, 2, 3, 4); in XLA_TEST_F() local
1726 input_arr.FillIota(0); in XLA_TEST_F()
1731 AddParam(LiteralUtil::CreateR4FromArray4D<float>(input_arr), &builder); in XLA_TEST_F()
1741 Array3D<float> input_arr(2, 6, 2); in XLA_TEST_F() local
[all …]
Dexhaustive_unary_test_f64.cc46 absl::Span<double> input_arr = (*input_literal)[0].data<double>(); in FillInput() local
50 input_arr[i] = this->ConvertAndReplaceKnownIncorrectValueWith(bits, 1); in FillInput()
Dexhaustive_unary_test_complex.cc69 absl::Span<NativeT> input_arr = (*input_literal)[0].data<NativeT>(); in FillInput() local
74 input_arr[i] = in FillInput()
Dexhaustive_unary_test_f32_or_smaller.cc204 absl::Span<NativeT> input_arr = (*input_literal)[0].data<NativeT>(); in FillInput() local
207 input_arr[i] = in FillInput()
Dreduce_test.cc897 std::vector<T> input_arr(num_elems, std::numeric_limits<T>::lowest()); in DoTest() local
898 auto input_literal = LiteralUtil::CreateR1<T>(input_arr); in DoTest()
/external/tensorflow/tensorflow/python/keras/
Dmodels_test.py340 input_arr = np.random.random((1, 3)).astype(np.float32)
344 clone(input_arr)
348 clone(input_arr)
353 feed_dict = {clone.input: input_arr}
355 self.assertAllClose(np.sum(input_arr), loss)
/external/tensorflow/tensorflow/lite/micro/kernels/ceva/
Dceva_tflm_lib.h73 void CEVA_TFLM_RoundToNearest_asm(const float* input_arr, float* output_arr,