| /external/tensorflow/tensorflow/lite/python/ |
| D | lite_test.py | 88 in_tensor = array_ops.placeholder(
90 out_tensor = in_tensor + in_tensor
94 converter = lite.TFLiteConverter.from_session(sess, [in_tensor],
118 in_tensor = array_ops.placeholder(shape=[4], dtype=dtypes.string)
119 out_tensor = array_ops.reshape(in_tensor, shape=[2, 2])
123 converter = lite.TFLiteConverter.from_session(sess, [in_tensor],
244 in_tensor = array_ops.placeholder(dtype=dtypes.float32)
245 out_tensor = in_tensor + in_tensor
249 converter = lite.TFLiteConverter.from_session(sess, [in_tensor],
258 in_tensor = array_ops.placeholder(dtype=dtypes.float32, shape=[])
[all …]
|
| D | convert_saved_model_test.py | 44 in_tensor = array_ops.placeholder(
46 _ = in_tensor + in_tensor
55 in_tensor = array_ops.placeholder(
57 _ = in_tensor + in_tensor
124 in_tensor = array_ops.placeholder(shape=shape, dtype=dtypes.float32)
125 out_tensor = in_tensor + in_tensor
126 inputs = {"x": in_tensor}
296 in_tensor = array_ops.placeholder(shape=[1, 28, 28], dtype=dtypes.float32)
297 out_tensor = in_tensor + in_tensor
298 sig_input_tensor = saved_model.utils.build_tensor_info(in_tensor)
|
| D | convert_test.py | 42 in_tensor = array_ops.placeholder(shape=[1, 16, 16, 3],
44 out_tensor = in_tensor + in_tensor
48 tflite_model = convert.toco_convert(sess.graph_def, [in_tensor],
59 in_tensor = array_ops.placeholder(shape=[4], dtype=dtypes.float32)
62 value=in_tensor, num_or_size_splits=[1, 1, 1, 1], axis=0)
70 in_tensor = array_ops.placeholder(shape=[1, 16, 16, 3],
72 out_tensor = array_ops.fake_quant_with_min_max_args(in_tensor + in_tensor,
77 sess.graph_def, [in_tensor], [out_tensor],
83 in_tensor = array_ops.placeholder(
86 in_tensor + in_tensor, min=0., max=1.)
[all …]
|
| D | lite_flex_test.py | 34 in_tensor = array_ops.placeholder(
36 out_tensor = in_tensor + in_tensor
40 converter = lite.TFLiteConverter.from_session(sess, [in_tensor],
|
| /external/tensorflow/tensorflow/contrib/fused_conv/python/ops/ |
| D | fused_conv2d_bias_activation_op_test_base.py | 639 def _NchwVectCToNchw(in_tensor): argument
641 t = array_ops.transpose(in_tensor, [0, 1, 4, 2, 3])
642 n = in_tensor.shape.dims[0].value
643 c = in_tensor.shape.dims[1].value * in_tensor.shape.dims[4].value
644 h = in_tensor.shape.dims[2].value
645 w = in_tensor.shape.dims[3].value
649 def _OihwVectIToHwio(in_tensor): argument
651 t = array_ops.transpose(in_tensor, [2, 3, 1, 4, 0])
652 o = in_tensor.shape.dims[0].value
653 i = in_tensor.shape.dims[1].value * in_tensor.shape.dims[4].value
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cast_op.h | 35 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>(); \
|
| D | mkl_transpose_op.cc | 124 Status MKLTransposeND(OpKernelContext* context, const Tensor& in_tensor, in MKLTransposeND() argument
131 memory::dims in_dims = TFShapeToMklDnnDims(in_tensor.shape()); in MKLTransposeND()
138 in.SetUsrMem(in_dims, in_strides, &in_tensor); in MKLTransposeND()
|
| /external/tensorflow/tensorflow/lite/toco/python/ |
| D | toco_from_protos_test.py | 37 def _run(self, sess, in_tensor, out_tensor, should_succeed): argument
56 input_array.name = TensorName(in_tensor)
57 input_array.shape.dims.extend(map(int, in_tensor.shape))
|
| /external/tensorflow/tensorflow/lite/testing/model_coverage/ |
| D | model_coverage_lib_test.py | 49 in_tensor = array_ops.placeholder(
51 _ = in_tensor + in_tensor
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops_test.py | 61 in_tensor = constant_op.constant(t0 if cpu == 0 else t1)
64 in_tensor, group_size, group_key, instance, 'Add', 'Div'))
|
| /external/tensorflow/tensorflow/contrib/kernel_methods/ |
| D | README.md | 16 out_tensor = kernel_mapper.map(in_tensor)
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter_test.cc | 1509 TfLiteTensor* in_tensor = &context->tensors[node->inputs->data[0]]; in CancelOpRegistration() local
1511 TfLiteIntArray* new_size = TfLiteIntArrayCopy(in_tensor->dims); in CancelOpRegistration()
1529 TfLiteTensor* in_tensor = &context->tensors[node->inputs->data[0]]; in OkOpRegistration() local
1531 TfLiteIntArray* new_size = TfLiteIntArrayCopy(in_tensor->dims); in OkOpRegistration()
|
| /external/tensorflow/tensorflow/contrib/quantize/python/ |
| D | fold_batch_norms.py | 940 def _AssertShapesMatch(op_name, in_tensor, out_tensor): argument
951 in_shape = in_tensor.get_shape()
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tensor_tracer.py | 559 for op in op.control_inputs + [in_tensor.op for in_tensor in op.inputs]:
|