Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/python/
Dlite_test.py121 in_tensor = array_ops.placeholder(
123 out_tensor = math_ops.add(in_tensor, in_tensor, name='add')
134 converter = lite.TFLiteConverter(frozen_graph_def, [in_tensor],
151 in_tensor = array_ops.placeholder(
153 math_ops.add(in_tensor, in_tensor, name='add')
208 in_tensor = array_ops.placeholder(
210 math_ops.add(in_tensor, in_tensor, name='add')
229 in_tensor = array_ops.placeholder(
231 out_tensor = in_tensor + in_tensor
235 converter = lite.TFLiteConverter.from_session(sess, [in_tensor],
[all …]
Dlite_flex_test.py52 in_tensor = array_ops.placeholder(shape=[1, 4], dtype=dtypes.float32)
53 out_tensor = in_tensor + in_tensor
57 converter = lite.TFLiteConverter.from_session(sess, [in_tensor],
81 in_tensor = array_ops.placeholder(
83 out_tensor = nn_ops.l2_loss(in_tensor)
84 converter = lite.TFLiteConverter.from_session(sess, [in_tensor],
94 in_tensor = array_ops.placeholder(shape=[1, 4], dtype=dtypes.float32)
95 out_tensor = in_tensor + in_tensor
99 converter = lite.TFLiteConverter.from_session(sess, [in_tensor],
216 in_tensor = array_ops.placeholder(
[all …]
Dtflite_convert_test.py117 in_tensor = array_ops.placeholder(
119 _ = in_tensor + in_tensor
138 in_tensor = array_ops.placeholder(
140 _ = in_tensor + in_tensor
157 in_tensor = random_ops.random_normal(shape=[1, 16, 16, 3], name='random')
158 _ = in_tensor + in_tensor
230 in_tensor = array_ops.placeholder(
232 out_tensor = in_tensor + in_tensor
233 inputs = {'x': in_tensor}
250 in_tensor = array_ops.placeholder(
[all …]
Dconvert_saved_model_test.py42 in_tensor = array_ops.placeholder(shape=shape, dtype=dtypes.float32)
43 out_tensor = in_tensor + in_tensor
44 inputs = {"x": in_tensor}
215 in_tensor = array_ops.placeholder(shape=[1, 28, 28], dtype=dtypes.float32)
216 out_tensor = in_tensor + in_tensor
217 sig_input_tensor = saved_model.utils.build_tensor_info(in_tensor)
Dconvert_test.py37 in_tensor = array_ops.placeholder(
39 out_tensor = in_tensor + in_tensor
44 sess.graph_def, input_tensors=[in_tensor], output_tensors=[out_tensor])
49 in_tensor = array_ops.placeholder(
52 in_tensor + in_tensor, min=0., max=1.)
57 input_tensors=[in_tensor],
65 in_tensor = array_ops.placeholder(
67 _ = in_tensor + in_tensor
Dutil_test.py69 in_tensor = array_ops.placeholder(dtype=dtypes.float32, shape=[4])
71 value=in_tensor, num_or_size_splits=[1, 1, 1, 1], axis=0)
155 in_tensor = array_ops.placeholder(
157 _ = in_tensor + in_tensor
165 in_tensor = array_ops.placeholder(
167 _ = in_tensor + in_tensor
Dmetrics_nonportable_test.py131 in_tensor = array_ops.placeholder(
133 math_ops.add(in_tensor, in_tensor, name='add')
453 in_tensor = array_ops.placeholder(
455 out_tensor = in_tensor + in_tensor
456 inputs = {'x': in_tensor}
Dlite_v2_test.py1434 in_tensor = tf.compat.v1.placeholder(
1437 out_tensor = in_tensor + constant
1438 inputs = {'x': in_tensor}
1488 in_tensor = tf.compat.v1.placeholder(
1491 assert_op = tf.Assert(tf.less_equal(in_tensor, constant), [in_tensor])
1493 out_tensor = in_tensor + constant
1494 inputs = {'x': in_tensor}
2385 in_tensor = tf.compat.v1.placeholder(
2387 out_tensor = in_tensor + in_tensor
2388 inputs = {'input': in_tensor}
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Dcollective_ops_multi_worker_test.py177 in_tensor = constant_op.constant([1.])
183 in_tensor,
201 in_tensor,
211 in_tensor,
223 in_tensor,
238 in_tensor = constant_op.constant([1.])
253 collective_ops.all_reduce(in_tensor, group_size, group_key,
259 collective_ops.all_reduce(in_tensor, group_size, group_key,
267 collective_ops.all_reduce(in_tensor, group_size, group_key, instance_key)
278 in_tensor = constant_op.constant([1.])
[all …]
Dcollective_ops_test.py377 in_tensor = constant_op.constant(in_value)
386 in_tensor,
396 in_tensor,
411 in_tensor = constant_op.constant(in_value)
420 in_tensor,
433 in_tensor,
738 in_tensor = constant_op.constant([1.])
757 in_tensor,
768 in_tensor,
784 in_tensor,
[all …]
/external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/
Dxla_sharding_test.py83 in_tensor = array_ops.ones([4, 5, 6], dtype=dtypes.float32)
85 self.assertAllEqual(in_tensor, result)
101 in_tensor = array_ops.ones([4, 5, 6], dtype=dtypes.float32)
103 self.assertAllEqual(in_tensor, result)
118 in_tensor = array_ops.ones([4, 5, 6], dtype=dtypes.float32)
121 self.assertAllEqual(in_tensor, result)
158 in_tensor = array_ops.ones([4, 5, 6], dtype=dtypes.float32)
161 self.assertAllEqual(in_tensor, result)
/external/tensorflow/tensorflow/core/kernels/
Dcast_op.h35 typename TTypes<IN_OUT>::ConstFlat in_tensor, \
39 in_tensor.unaryExpr(LSBZeroSetter<IN_OUT, OUT_TYPE>()) \
42 out_tensor.device(d) = in_tensor.template cast<OUT_TYPE>(); \
53 typename TTypes<IN_OUT>::ConstFlat in_tensor, \
57 in_tensor.unaryExpr(LSBZeroSetter<IN_OUT, OUT_TYPE>()) \
60 out_tensor.device(d) = in_tensor.template cast<OUT_TYPE>(); \
80 typename TTypes<IN_OUT>::ConstFlat in_tensor, \
82 out_tensor.device(d) = in_tensor.template cast<OUT_TYPE>(); \
100 typename TTypes<IN_OUT>::ConstFlat in_tensor, \
102 out_tensor.device(d) = in_tensor.template cast<OUT_TYPE>(); \
/external/ComputeLibrary/src/dynamic_fusion/sketch/utils/
DDependencyGraph.h194 for(auto in_tensor : inputs) variable
198 link_input(op, in_tensor);
368 void link_input(OperatorId op, TensorId in_tensor) in link_input() argument
371 if(!tensor_exists(in_tensor)) in link_input()
373 insert_new_tensor(in_tensor); in link_input()
375 ARM_COMPUTE_ERROR_ON(are_connected(op, in_tensor)); // Prevent repetitive linking in link_input()
376 _adj_src_tensors[op].push_back(in_tensor); in link_input()
377 _adj_dst_ops[in_tensor].push_back(op); in link_input()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_transpose_op.cc65 Status MKLTransposeND(OpKernelContext* context, const Tensor& in_tensor, in MKLTransposeND() argument
72 memory::dims in_dims = TFShapeToMklDnnDims(in_tensor.shape()); in MKLTransposeND()
80 in.SetUsrMem(in_dims, in_strides, &in_tensor); in MKLTransposeND()
89 in.SetUsrMemDataHandle(&in_tensor, transpose_stream); in MKLTransposeND()
/external/tensorflow/tensorflow/lite/toco/python/
Dtoco_from_protos_test.py33 def _run(self, sess, in_tensor, out_tensor, should_succeed): argument
52 input_array.name = TensorName(in_tensor)
53 input_array.shape.dims.extend(map(int, in_tensor.shape))
/external/tensorflow/tensorflow/python/ops/
Dcollective_ops_gpu_test.py290 in_tensor = constant_op.constant(1.)
297 in_tensor,
318 in_tensor,
329 in_tensor,
Dcollective_ops_test.py99 in_tensor = constant_op.constant(t0 if cpu == 0 else t1)
102 in_tensor, group_size, group_key, instance, 'Add', 'Div'))
Dimage_grad_test_base.py144 def func(in_tensor): argument
146 in_tensor,
/external/tensorflow/tensorflow/lite/python/metrics/
Dmetrics_nonportable_test.py131 in_tensor = array_ops.placeholder(
133 math_ops.add(in_tensor, in_tensor, name='add')
455 in_tensor = array_ops.placeholder(
457 out_tensor = in_tensor + in_tensor
458 inputs = {'x': in_tensor}
/external/tensorflow/tensorflow/python/kernel_tests/linalg/
Dmatrix_exponential_op_test.py138 in_tensor = [[np.inf, 1.], [1., 1.]]
139 result = self.evaluate(linalg_impl.matrix_exponential(in_tensor))
/external/tensorflow/tensorflow/python/tpu/
Dtensor_tracer_report.py82 for op in op.control_inputs + [in_tensor.op for in_tensor in op.inputs]:
/external/tensorflow/tensorflow/lite/
Dinterpreter_test.cc1479 const TfLiteTensor* in_tensor; in CancelOpRegistration() local
1480 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 0, &in_tensor)); in CancelOpRegistration()
1483 TfLiteIntArray* new_size = TfLiteIntArrayCopy(in_tensor->dims); in CancelOpRegistration()
1501 const TfLiteTensor* in_tensor; in OkOpRegistration() local
1502 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 0, &in_tensor)); in OkOpRegistration()
1505 TfLiteIntArray* new_size = TfLiteIntArrayCopy(in_tensor->dims); in OkOpRegistration()