Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/kernel_tests/
Dsparse_reshape_op_test.py101 output_val = self.evaluate(sp_output)
102 self.assertAllEqual(output_val.indices, input_val.indices)
103 self.assertAllEqual(output_val.values, input_val.values)
104 self.assertAllEqual(output_val.dense_shape, input_val.dense_shape)
113 output_val = sess.run(sp_output, {sp_input: input_val})
114 self.assertAllEqual(output_val.indices, input_val.indices)
115 self.assertAllEqual(output_val.values, input_val.values)
116 self.assertAllEqual(output_val.dense_shape, input_val.dense_shape)
126 output_val = sess.run(sp_output, {sp_input: input_val})
127 self.assertAllEqual(output_val.indices, input_val.indices)
[all …]
Dsparse_reorder_op_test.py64 output_val = self.evaluate(sp_output)
65 self.assertAllEqual(output_val.indices, input_val.indices)
66 self.assertAllEqual(output_val.values, input_val.values)
67 self.assertAllEqual(output_val.dense_shape, input_val.dense_shape)
76 output_val = sess.run(sp_output, {sp_input: input_val})
77 self.assertAllEqual(output_val.indices, input_val.indices)
78 self.assertAllEqual(output_val.values, input_val.values)
79 self.assertAllEqual(output_val.dense_shape, input_val.dense_shape)
88 output_val = self.evaluate(sp_output)
89 self.assertAllEqual(output_val.indices, expected_output_val.indices)
[all …]
/external/tensorflow/tensorflow/python/grappler/
Dlayout_optimizer_test.py290 output_val = sess.run(output, run_metadata=metadata)
306 self.assertAllClose(output_val_ref, output_val, atol=1e-3)
328 output_val = sess.run(output, run_metadata=metadata, feed_dict={dim: 3})
342 self.assertAllClose(output_val_ref, output_val, atol=1e-3)
361 output_val = sess.run(output, run_metadata=metadata, feed_dict={dim: 3})
377 self.assertAllClose(output_val_ref, output_val, atol=1e-3)
396 output_val = sess.run(output, run_metadata=metadata)
412 self.assertAllClose(output_val_ref, output_val, atol=1e-3)
428 output_val = sess.run(output, run_metadata=metadata)
442 self.assertAllClose(output_val_ref, output_val, atol=1e-3)
[all …]
Dauto_mixed_precision_test.py366 output_val = sess.run(fetches, run_metadata=metadata)
368 return output_val_ref, output_val, metadata.cost_graph
414 output_val_ref, output_val, cost_graph = self._run(mode, a)
430 self.assertAllClose(output_val_ref, output_val, atol=2e-2, rtol=2e-2)
432 self.assertAllClose(output_val_ref, output_val, atol=2e-3, rtol=1e-3)
445 output_val_ref, output_val, cost_graph = self._run(mode, output)
463 self.assertAllClose(output_val_ref, output_val, atol=tol, rtol=tol)
479 output_val_ref, output_val, cost_graph = self._run(mode, output)
488 self.assertAllClose(output_val_ref, output_val, atol=1e-2, rtol=1e-2)
508 output_val_ref, output_val, cost_graph = self._run(mode, output)
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dtanh.h97 uint8_t output_val; in Tanh() local
99 output_val = 0; in Tanh()
101 output_val = 255; in Tanh()
120 output_val = static_cast<uint8_t>(output_val_s32); in Tanh()
122 output_data[i] = output_val; in Tanh()
Dl2normalization.h80 int32_t output_val = in L2Normalization() local
83 output_data[depth * i + c] = static_cast<uint8_t>(output_val); in L2Normalization()
Dlegacy_reference_ops.h801 uint8 output_val; in Logistic() local
803 output_val = 0; in Logistic()
805 output_val = 255; in Logistic()
823 output_val = static_cast<uint8>(output_val_s32); in Logistic()
825 output_data[i] = output_val; in Logistic()
/external/tensorflow/tensorflow/python/ops/
Dsparse_ops.py419 output_ind, output_val, output_shape = (
440 return sparse_tensor.SparseTensor(output_ind, output_val, output_shape)
445 output = sparse_tensor.SparseTensor(output_ind, output_val, output_shape)
583 output_ind, output_val, output_shape = (
595 return sparse_tensor.SparseTensor(output_ind, output_val, output_shape)
1311 output_ind, output_val, output_shape = (
1320 return sparse_tensor.SparseTensor(output_ind, output_val, output_shape)
1462 output_ind, output_val, output_shape = (
1467 return sparse_tensor.SparseTensor(output_ind, output_val, output_shape)
1530 output_ind, output_val, output_shape = (
[all …]
/external/XNNPACK/test/
Ddwconv2d-microkernel-tester.h225 for (float& output_val : output_ref) { variable
226 output_val = std::max(std::min(output_val, output_max), output_min);
Ddwconv-microkernel-tester.h460 for (float& output_val : output_ref) { variable
461 output_val = std::max(std::min(output_val, output_max), output_min);
626 for (float& output_val : output_ref) { variable
627 output_val = std::max(std::min(output_val, output_max), output_min);
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dconvert_tfl_uint8.cc216 for (Value output_val : op.getResults()) { in convert_graph_uint8_tensor() local
219 output_val.getType().dyn_cast<mlir::RankedTensorType>(); in convert_graph_uint8_tensor()
254 output_val.setType(new_type); in convert_graph_uint8_tensor()
Dlegalize_common.h59 Operation* op, Value output_val,
Dlegalize_common.cc441 Operation* op, Value output_val, in convertMultiplyOp() argument
449 output_val.getType().dyn_cast<RankedTensorType>(); in convertMultiplyOp()
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dxla_control_flow_ops_test.py126 output_val, ans_val = self.evaluate([output, ans])
127 self.assertAllClose(ans_val, output_val)
Dcontrol_flow_ops_test.py236 output_val, ans_val = self.evaluate([output, ans])
237 self.assertAllClose(ans_val, output_val)
250 output_val, ans_val = self.evaluate([output, ans])
251 self.assertAllClose(ans_val, output_val)
262 output_val, ans_val = self.evaluate([output, ans])
263 self.assertAllClose(ans_val, output_val)
274 output_val, ans_val = self.evaluate([output, ans])
275 self.assertAllClose(ans_val, output_val)
291 output_val, ans_val = self.evaluate([output, ans])
292 self.assertAllClose(ans_val, output_val)
[all …]
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Doptimize.cc235 DenseElementsAttr GetShape(Value output_val) { in GetShape() argument
236 auto output_type = output_val.getType().cast<RankedTensorType>(); in GetShape()
246 mlir::IntegerType::get(output_val.getContext(), 32)), in GetShape()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h1789 int32 output_val = std::min(255, std::max(0, unclamped_output_val)); in L2Normalization() local
1790 *output_data = static_cast<uint8>(output_val); in L2Normalization()
7275 uint8 output_val; in Tanh16bitPrecision() local
7277 output_val = 0; in Tanh16bitPrecision()
7279 output_val = 255; in Tanh16bitPrecision()
7297 output_val = static_cast<uint8>(output_val_s16); in Tanh16bitPrecision()
7299 output_data[c] = output_val; in Tanh16bitPrecision()
7369 int8 output_val; in Tanh16bitPrecision() local
7371 output_val = -128; in Tanh16bitPrecision()
7373 output_val = 127; in Tanh16bitPrecision()
[all …]
Dneon_tensor_utils.cc872 const int16x8_t output_val = vld1q_s16(output + i); in NeonMatrixBatchVectorAccumulateImpl() local
873 const int32x4_t first_half = vmovl_s16(vget_low_s16(output_val)); in NeonMatrixBatchVectorAccumulateImpl()
874 const int32x4_t second_half = vmovl_s16(vget_high_s16(output_val)); in NeonMatrixBatchVectorAccumulateImpl()
928 const int8x16_t output_val = vld1q_s8(output + i); in NeonMatrixBatchVectorAccumulateImpl() local
929 const int16x8_t first_half = vmovl_s8(vget_low_s8(output_val)); in NeonMatrixBatchVectorAccumulateImpl()
930 const int16x8_t second_half = vmovl_s8(vget_high_s8(output_val)); in NeonMatrixBatchVectorAccumulateImpl()
Dlegacy_optimized_ops.h4379 uint8 output_val; in Logistic() local
4381 output_val = 0; in Logistic()
4383 output_val = 255; in Logistic()
4399 output_val = static_cast<uint8>(output_val_s32); in Logistic()
4401 output_data[c] = output_val; in Logistic()
4581 uint8 output_val; in Tanh() local
4583 output_val = 0; in Tanh()
4585 output_val = 255; in Tanh()
4602 output_val = static_cast<uint8>(output_val_s32); in Tanh()
4604 output_data[c] = output_val; in Tanh()
/external/tensorflow/tensorflow/python/keras/engine/
Dfunctional_test.py805 output_val = m.predict(x_val)
814 self.assertAllClose(output_val, output_val_2, atol=1e-6)
833 output_val = m.predict(x_val)
842 self.assertAllClose(output_val, output_val_2, atol=1e-6)
/external/tensorflow/tensorflow/python/framework/
Dfunction_test.py909 output_val = sess.run(
911 self.assertEqual(output_val.shape, (32, 100))
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column_v2_test.py1146 output_val = self.evaluate(output)
1148 output_val.indices)
1149 for val in output_val.values:
1151 self.assertAllEqual([2, 4], output_val.dense_shape)
Dfeature_column_test.py1030 output_val = self.evaluate(output)
1032 [[0, 0], [0, 1], [1, 0], [1, 1], [1, 2], [1, 3]], output_val.indices)
1033 for val in output_val.values:
1035 self.assertAllEqual([2, 4], output_val.dense_shape)