Searched refs:tf_tensor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_input_conversion_op.cc | 191 const Tensor* tf_tensor; in Compute() local
198 tf_tensor = &input_tensor_1; in Compute()
204 tf_tensor = &input_tensor_0; in Compute()
214 tf_tensor->shape().num_elements()) { in Compute()
241 tf_input.SetUsrMem(input_tf_md, tf_tensor); in Compute()
254 CHECK(tensor_out->CopyFrom(*tf_tensor, tensor_out->shape())); in Compute()
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | ndarray_tensor.cc | 471 Safe_TF_TensorPtr tf_tensor = make_safe(static_cast<TF_Tensor*>(nullptr)); in NdarrayToTensor() local
472 Status s = PyArrayToTF_Tensor(obj, &tf_tensor); in NdarrayToTensor()
476 return TF_TensorToTensor(tf_tensor.get(), ret); in NdarrayToTensor()
481 Safe_TF_TensorPtr tf_tensor = make_safe(TF_TensorFromTensor(t, status)); in TensorToNdarray() local
487 return TF_TensorToPyArray(std::move(tf_tensor), ret); in TensorToNdarray()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | test_util.py | 2493 def assertShapeEqual(self, np_array, tf_tensor, msg=None): argument
2506 if not isinstance(tf_tensor, ops.Tensor):
2509 np_array.shape, tf_tensor.get_shape().as_list(), msg=msg)
|