Home
last modified time | relevance | path

Searched refs:tensor_x (Results 1 – 4 of 4) sorted by relevance

/external/tflite-support/tensorflow_lite_support/cc/task/vision/
Dimage_segmenter.cc351 int tensor_x; in Postprocess() local
370 /*to_x=*/&tensor_x, in Postprocess()
376 GetOutputConfidence(*output_tensor, tensor_x, tensor_y, d); in Postprocess()
399 /*to_x=*/&tensor_x, in Postprocess()
403 GetOutputConfidence(*output_tensor, tensor_x, tensor_y, d)); in Postprocess()
/external/tensorflow/tensorflow/python/autograph/operators/
Ddata_structures.py201 tensor_x = ops.convert_to_tensor(x)
203 element_shape=array_ops.shape(tensor_x),
204 element_dtype=tensor_x.dtype)
/external/tensorflow/tensorflow/python/framework/
Dtest_util_test.py724 tensor_x = random_ops.random_uniform((5, 2, 1))
728 self.assertShapeEqual(tensor_x, tensor_y)
730 self.assertShapeEqual(tensor_x, tensor_z)
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.cc3529 TF_ASSIGN_OR_RETURN(auto tensor_x, in GetCudnnOperationGraph()
3586 .setxDesc(tensor_x) in GetCudnnOperationGraph()
3603 VLOG(4) << "\nTensor_x: " << tensor_x.describe() in GetCudnnOperationGraph()
3664 TF_ASSIGN_OR_RETURN(auto tensor_x, in GetCudnnFusedOperationGraph()
3758 .setxDesc(tensor_x) in GetCudnnFusedOperationGraph()
3874 VLOG(4) << "\nTensor_x: " << tensor_x.describe() in GetCudnnFusedOperationGraph()