Searched defs:tf_tensor (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/c/kernels/ |
| D | tensor_shape_utils_test.cc | 32 TF_Tensor* tf_tensor; member
40 TF_Tensor* tf_tensor = TF_TensorFromTensor(tensor, &status); in TestShapeMatch() local
|
| /external/tensorflow/tensorflow/python/dlpack/ |
| D | dlpack.py | 27 def to_dlpack(tf_tensor): argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | eval_util.cc | 109 TF_Tensor* tf_tensor = TF_TensorFromTensor(tensor, &status->status); in EvaluateOperation() local
135 TF_Tensor* tf_tensor = TFE_TensorHandleResolve(tensor_handle, status); in EvaluateOperation() local
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | core.cc | 57 TF_Tensor* tf_tensor = TF_TensorFromTensor(*t, status); in Resolve() local
95 TF_Tensor* tf_tensor = TF_TensorFromTensor(tensor, status); in Resolve() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_input_conversion_op.cc | 193 const Tensor* tf_tensor; in Compute() local
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | ndarray_tensor.cc | 552 Safe_TF_TensorPtr tf_tensor = make_safe(static_cast<TF_Tensor*>(nullptr)); in NdarrayToTensor() local
562 Safe_TF_TensorPtr tf_tensor = make_safe(TF_TensorFromTensor(t, &status)); in TensorToNdarray() local
|
| D | py_seq_tensor.cc | 688 Safe_TF_TensorPtr tf_tensor = make_safe(static_cast<TF_Tensor*>(nullptr)); in NumpyToTFE_TensorHandle() local
|
| /external/tensorflow/tensorflow/c/ |
| D | kernels.cc | 475 TF_Tensor* tf_tensor = TF_TensorFromTensor(*tensor, &s); in TF_AllocateOutput() local
543 TF_Tensor* tf_tensor; in TF_AllocateTemp() local
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel.cc | 632 const tensorflow::Tensor& tf_tensor = buffer_map->GetTensor(tensor_index); in Eval() local
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | test_util.py | 3103 def assertShapeEqual(self, np_array, tf_tensor, msg=None): argument
|