Home
last modified time | relevance | path

Searched refs:tensor_type (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/lite/kernels/
Dadd_test.cc193 template <TensorType tensor_type, typename integer_dtype>
203 QuantizedAddOpModel m({tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, in QuantizedTestsNoActivation()
204 {tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, in QuantizedTestsNoActivation()
205 {tensor_type, {}, -1.0, 1.0}, in QuantizedTestsNoActivation()
250 template <enum TensorType tensor_type, typename integer_dtype>
260 QuantizedAddOpModel m({tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, in QuantizedTestsActivationRELU_N1_TO_1()
261 {tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, in QuantizedTestsActivationRELU_N1_TO_1()
262 {tensor_type, {}, -1.0, 1.0}, in QuantizedTestsActivationRELU_N1_TO_1()
282 template <enum TensorType tensor_type, typename integer_dtype>
288 QuantizedAddOpModel m({tensor_type, test_shapes[i], -3.0, 3.0}, in QuantizedVariousInputShapes()
[all …]
Dsub_test.cc196 template <TensorType tensor_type, typename integer_dtype>
207 QuantizedSubOpModel m({tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, in QuantizedTestsNoActivation()
208 {tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, in QuantizedTestsNoActivation()
209 {tensor_type, {}, -1.0, 1.0}, in QuantizedTestsNoActivation()
229 template <TensorType tensor_type, typename integer_dtype>
239 QuantizedSubOpModel m({tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, in QuantizedTestsActivationRELU_N1_TO_1()
240 {tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, in QuantizedTestsActivationRELU_N1_TO_1()
241 {tensor_type, {}, -1.0, 1.0}, in QuantizedTestsActivationRELU_N1_TO_1()
260 template <TensorType tensor_type, typename integer_dtype>
266 QuantizedSubOpModel m({tensor_type, test_shapes[i], -3.0, 3.0}, in QuantizedVariousInputShapes()
[all …]
Dmul_test.cc195 template <TensorType tensor_type, typename integer_dtype>
197 QuantizedMulOpModel m({tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, in NoActivation()
198 {tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, in NoActivation()
199 {tensor_type, {}, -1.0, 1.0}, in NoActivation()
232 template <TensorType tensor_type, typename integer_dtype>
240 {tensor_type, {}, kMinUint8, kMaxUint8}, in NoActivationInt16With8BitOutput()
265 template <TensorType tensor_type, typename integer_dtype>
271 QuantizedMulOpModel m({tensor_type, test_shapes[i], -3.0, 3.0}, in WithBroadcast()
272 {tensor_type, {}, -3.0, 3.0}, // always a scalar in WithBroadcast()
273 {tensor_type, {}, -3.0, 3.0}, in WithBroadcast()
Dl2norm_test.cc29 const TensorType tensor_type, in L2NormOpModel() argument
31 TensorData data = TensorData{tensor_type}; in L2NormOpModel()
32 if (tensor_type != TensorType_FLOAT32) { in L2NormOpModel()
39 if (tensor_type != TensorType_FLOAT32) { in L2NormOpModel()
Dsvdf_test.cc206 int rank, TensorType tensor_type) in HybridSVDFOpModel() argument
208 tensor_type, tensor_type) { in HybridSVDFOpModel()
209 tensor_type_ = tensor_type; in HybridSVDFOpModel()
Dbasic_rnn_test.cc236 HybridRNNOpModel(int batches, int units, int size, TensorType tensor_type) in HybridRNNOpModel() argument
237 : RNNOpModel(batches, units, size, tensor_type, tensor_type) { in HybridRNNOpModel()
238 tensor_type_ = tensor_type; in HybridRNNOpModel()
Dunidirectional_sequence_rnn_test.cc252 TensorType tensor_type) in HybridUnidirectionalRNNOpModel() argument
254 tensor_type, tensor_type) { in HybridUnidirectionalRNNOpModel()
255 tensor_type_ = tensor_type; in HybridUnidirectionalRNNOpModel()
Dlstm_test.cc276 TensorType tensor_type) in HybridLSTMOpModel() argument
279 proj_clip, input_shapes, tensor_type) { in HybridLSTMOpModel()
280 tensor_type_ = tensor_type; in HybridLSTMOpModel()
1653 TensorType tensor_type) in HybridLayerNormLSTMOpModel() argument
1657 input_shapes, tensor_type) { in HybridLayerNormLSTMOpModel()
1658 tensor_type_ = tensor_type; in HybridLayerNormLSTMOpModel()
Dunidirectional_sequence_lstm_test.cc247 TensorType tensor_type) in HybridUnidirectionalLSTMOpModel() argument
251 cell_clip, proj_clip, input_shapes, tensor_type) { in HybridUnidirectionalLSTMOpModel()
252 tensor_type_ = tensor_type; in HybridUnidirectionalLSTMOpModel()
/external/tensorflow/tensorflow/lite/python/
Dinterpreter.py130 tensor_type = self._interpreter.TensorType(tensor_index)
133 if not tensor_name or not tensor_type:
140 'dtype': tensor_type,
/external/tensorflow/tensorflow/core/kernels/
Dload_and_remap_matrix_op.cc137 DataType tensor_type; in Compute() local
140 old_tensor_name, &tensor_type, &tensor_shape)); in Compute()
141 OP_REQUIRES(context, tensor_type == DT_FLOAT, in Compute()
144 DataTypeString(tensor_type), " instead of expected type ", in Compute()
/external/tensorflow/tensorflow/python/client/
Dsession.py140 tensor_type, argument
186 if issubclass(conversion_function[0], tensor_type):
188 tensor_type)
190 _REGISTERED_EXPANSIONS.insert(0, (tensor_type, fetch_function, feed_function,
268 for tensor_type, fetch_fn, _, _ in _REGISTERED_EXPANSIONS:
269 if isinstance(fetch, tensor_type):
1011 for tensor_type, _, _, feed_fn in _REGISTERED_EXPANSIONS:
1012 if isinstance(feed, tensor_type):
1068 for tensor_type, _, feed_fn, _ in _REGISTERED_EXPANSIONS:
1069 if isinstance(feed, tensor_type):
/external/tensorflow/tensorflow/python/kernel_tests/
Darray_ops_test.py497 def __init__(self, test, x, tensor_type=dtypes.int32, check_type_infer=True): argument
498 self.x_np = np.array(x).astype(tensor_type.as_numpy_dtype)
499 if tensor_type.is_bool:
502 if tensor_type.is_complex:
505 self.x = constant_op.constant(self.x_np, dtype=tensor_type)
560 for tensor_type in STRIDED_SLICE_TYPES:
563 self, StridedSliceChecker.REF_TENSOR, tensor_type=tensor_type)
581 checker2 = StridedSliceChecker(self, 5, tensor_type=tensor_type)
1029 def __init__(self, test, x, tensor_type=dtypes.float32, use_resource=False): argument
1030 self.tensor_type = tensor_type
[all …]
/external/tensorflow/tensorflow/lite/toco/tflite/
Dtypes.cc112 ArrayDataType DataType::Deserialize(int tensor_type) { in Deserialize() argument
113 switch (::tflite::TensorType(tensor_type)) { in Deserialize()
131 LOG(FATAL) << "Unhandled tensor type '" << tensor_type << "'."; in Deserialize()
Dtypes.h27 static ArrayDataType Deserialize(int tensor_type);
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc56 TfLiteStatus ConvertTensorType(TensorType tensor_type, TfLiteType* type, in ConvertTensorType() argument
58 switch (tensor_type) { in ConvertTensorType()
88 EnumNameTensorType(tensor_type), tensor_type); in ConvertTensorType()
Dflatbuffer_conversions.h63 TfLiteStatus ConvertTensorType(TensorType tensor_type, TfLiteType* type,
/external/tensorflow/tensorflow/python/ops/
Dio_ops.py75 def _restore_slice(file_pattern, tensor_name, shape_and_slice, tensor_type, argument
93 base_type = dtypes.as_dtype(tensor_type).base_dtype
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcluster_function_library_runtime_test.cc149 key: "tensor_type" in TEST_F()
190 key: "tensor_type" in TEST_F()
/external/tensorflow/tensorflow/contrib/bigtable/python/ops/
Dbigtable_api.py478 for tensor_type in nest.flatten(
480 if tensor_type != dtypes.string:
/external/tensorflow/tensorflow/python/util/
Dutil.cc279 PyObject* tensor_type = GetRegisteredType("Tensor"); in IsTensorHelper() local
280 if (TF_PREDICT_FALSE(tensor_type == nullptr)) { in IsTensorHelper()
289 return PyObject_IsInstance(to_check, tensor_type); in IsTensorHelper()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc149 auto tensor_type = tensor_ptr->dtype(); in GetTensorAddress() local
150 switch (tensor_type) { in GetTensorAddress()
155 LOG(ERROR) << "Unsupported Data type " << DataTypeString(tensor_type); in GetTensorAddress()
/external/tensorflow/tensorflow/contrib/mpi_collectives/
Dmpi_message.proto37 DataType tensor_type = 3; field
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc419 TfLiteType tensor_type = tensor->type; in AddTensor() local
420 if (hybrid_op && (tensor_type == kTfLiteUInt8)) { in AddTensor()
423 tensor_type = kTfLiteInt8; in AddTensor()
425 switch (tensor_type) { in AddTensor()
/external/tensorflow/tensorflow/python/framework/
Dops.py177 def register_dense_tensor_like_type(tensor_type): argument
190 if not isinstance(tensor_type.name, property):
192 tensor_type.__name__)
195 tensor_type.__name__)
197 if not isinstance(tensor_type.dtype, property):
199 tensor_type.__name__)
202 tensor_type.__name__)
207 _TENSOR_LIKE_TYPES = tuple(list(_TENSOR_LIKE_TYPES) + [tensor_type])

12