Home
last modified time | relevance | path

Searched refs:input_tensor (Results 1 – 25 of 28) sorted by relevance

12

/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/experimental/mlir/testing/op_tests/
A Dcontrol_dep.py37 input_tensor = tf.compat.v1.placeholder(
40 assert_op = tf.compat.v1.assert_greater_equal(input_tensor,
41 input_tensor - 1)
44 input_tensor, filter_value, strides=(1, 1, 1, 1), padding="SAME")
45 return [input_tensor], [out]
A Dconv_bias_activation.py69 def add_conv(input_tensor, filter_input, parameters): argument
71 input_tensor,
87 input_tensor = tf.compat.v1.placeholder(
92 input_tensors = [input_tensor]
95 out = add_conv(input_tensor, filter_input, parameters)
98 input_tensor,
A Didentify_dilated_conv.py55 input_tensor = tf.compat.v1.placeholder(
63 input_tensors = [input_tensor]
67 input_tensors = [input_tensor, filter_input]
72 input_tensor,
A Dsoftplus.py37 input_tensor = tf.compat.v1.placeholder(
42 out = tf.math.softplus(input_tensor)
43 return [input_tensor], [out]
A Dreal.py37 input_tensor = tf.compat.v1.placeholder(
41 out = tf.math.real(input_tensor)
42 return [input_tensor], [out]
A Dimag.py37 input_tensor = tf.compat.v1.placeholder(
41 out = tf.math.imag(input_tensor)
42 return [input_tensor], [out]
A Dreciprocal.py38 input_tensor = tf.compat.v1.placeholder(
43 out = tf.math.reciprocal(input_tensor)
44 return [input_tensor], [out]
A Dbroadcast_to.py52 input_tensor = tf.compat.v1.placeholder(
57 out = tf.broadcast_to(input_tensor, shape=parameters["output_shape"])
58 return [input_tensor], [out]
A Dcomplex_abs.py42 input_tensor = tf.compat.v1.placeholder(
46 out = tf.raw_ops.ComplexAbs(x=input_tensor, Tout=parameters["Tout"])
47 return [input_tensor], [out]
A Dconv3d.py42 input_tensor = tf.compat.v1.placeholder(
52 input_tensor,
57 return [input_tensor, filter_tensor], [out]
A Dmax_pool_with_argmax.py46 input_tensor = tf.compat.v1.placeholder(
49 input_tensor,
54 return [input_tensor], [updates, indices]
A Ddense_image_warp.py40 input_tensor = tf.compat.v1.placeholder(
44 output = dense_image_warp_annotated(input_tensor, flow_tensor)
45 return [input_tensor, flow_tensor], [output]
A Dshape_to_strided_slice.py33 input_tensor = tf.compat.v1.placeholder(
40 tensors = [input_tensor]
42 tf.shape(input_tensor),
A Ddynamic_rnn.py51 input_tensor = tf.placeholder(dtype=parameters["dtype"], shape=input_shape)
55 lstm_cell, input_tensor, dtype=parameters["dtype"])
56 return [input_tensor], [output]
A Didentify_dilated_conv1d.py53 input_tensor = tf.compat.v1.placeholder(
55 input_tensors = [input_tensor, filter_input]
58 input_tensor,
A Dis_finite.py40 input_tensor = tf.compat.v1.placeholder(
42 out = tf.math.is_finite(input_tensor)
43 return [input_tensor], [out]
/AliOS-Things-master/solutions/tflite_micro_speech_demo/micro_speech/train/speech_commands/
A Dfreeze.py176 def save_saved_model(file_name, sess, input_tensor, output_tensor): argument
188 'input': tf.compat.v1.saved_model.utils.build_tensor_info(input_tensor)
225 input_tensor, output_tensor = create_inference_graph(
240 save_saved_model(FLAGS.output_file, sess, input_tensor, output_tensor)
A Dfreeze_test.py115 input_tensor, output_tensor = freeze.create_inference_graph(
128 freeze.save_saved_model(saved_model_path, sess, input_tensor,
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/
A Dmicro_interpreter.h79 TfLiteTensor* input_tensor(size_t index) { return input(index); } in input_tensor() function
82 if (TfLiteTensor* tensor_ptr = input_tensor(tensor_index)) { in typed_input_tensor()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/
A Dconcatenation.cc139 const TfLiteTensor* input_tensor = GetInput(context, node, 0); in Prepare() local
140 TF_LITE_ENSURE(context, input_tensor != nullptr); in Prepare()
141 TfLiteType input_type = input_tensor->type; in Prepare()
A Dsvdf.h51 const TfLiteEvalTensor* input_tensor,
A Dsvdf_common.cc52 const TfLiteEvalTensor* input_tensor, in EvalIntegerSvdfReference() argument
61 const int n_batch = input_tensor->dims->data[0]; in EvalIntegerSvdfReference()
62 const int n_input = input_tensor->dims->data[1]; in EvalIntegerSvdfReference()
95 const int8_t* input = tflite::micro::GetTensorData<int8_t>(input_tensor); in EvalIntegerSvdfReference()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/experimental/mlir/testing/
A Dmlir_convert.py156 for input_tensor in input_tensors:
157 shape = input_tensor[1]
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/cmsis_nn/
A Dsvdf.cc207 const TfLiteEvalTensor* input_tensor, in EvalIntegerSVDF() argument
215 input_dims.n = input_tensor->dims->data[0]; in EvalIntegerSVDF()
216 input_dims.h = input_tensor->dims->data[1]; in EvalIntegerSVDF()
271 (int8_t*)tflite::micro::GetTensorData<int8_t>(input_tensor), &state_dims, in EvalIntegerSVDF()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/
A Dinterpreter.h360 TfLiteTensor* input_tensor(size_t index) { return tensor(inputs()[index]); } in input_tensor() function
364 const TfLiteTensor* input_tensor(size_t index) const { in input_tensor() function

Completed in 21 milliseconds

12