Home
last modified time | relevance | path

Searched defs:tensor_type (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/contrib/mpi_collectives/
Dmpi_message.proto37 DataType tensor_type = 3; field
/external/tensorflow/tensorflow/lite/kernels/
Dl2norm_test.cc29 const TensorType tensor_type, in L2NormOpModel()
Dbasic_rnn_test.cc236 HybridRNNOpModel(int batches, int units, int size, TensorType tensor_type) in HybridRNNOpModel()
Dsvdf_test.cc206 int rank, TensorType tensor_type) in HybridSVDFOpModel()
Dunidirectional_sequence_rnn_test.cc252 TensorType tensor_type) in HybridUnidirectionalRNNOpModel()
Dlstm_test.cc276 TensorType tensor_type) in HybridLSTMOpModel()
1653 TensorType tensor_type) in HybridLayerNormLSTMOpModel()
Dunidirectional_sequence_lstm_test.cc247 TensorType tensor_type) in HybridUnidirectionalLSTMOpModel()
/external/tensorflow/tensorflow/lite/toco/tflite/
Dtypes.cc112 ArrayDataType DataType::Deserialize(int tensor_type) { in Deserialize()
/external/tensorflow/tensorflow/core/kernels/
Dload_and_remap_matrix_op.cc137 DataType tensor_type; in Compute() local
/external/tensorflow/tensorflow/python/ops/
Dio_ops.py75 def _restore_slice(file_pattern, tensor_name, shape_and_slice, tensor_type, argument
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc56 TfLiteStatus ConvertTensorType(TensorType tensor_type, TfLiteType* type, in ConvertTensorType()
/external/tensorflow/tensorflow/python/kernel_tests/
Darray_ops_test.py497 def __init__(self, test, x, tensor_type=dtypes.int32, check_type_infer=True): argument
1029 def __init__(self, test, x, tensor_type=dtypes.float32, use_resource=False): argument
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc149 auto tensor_type = tensor_ptr->dtype(); in GetTensorAddress() local
/external/tensorflow/tensorflow/python/util/
Dutil.cc279 PyObject* tensor_type = GetRegisteredType("Tensor"); in IsTensorHelper() local
/external/tensorflow/tensorflow/python/client/
Dsession.py140 tensor_type, argument
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc419 TfLiteType tensor_type = tensor->type; in AddTensor() local
/external/tensorflow/tensorflow/lite/
Dinterpreter_test.cc192 TfLiteType tensor_type = kTfLiteInt8; in TEST() local
/external/tensorflow/tensorflow/python/framework/
Dops.py177 def register_dense_tensor_like_type(tensor_type): argument