Searched refs:FromTensor (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/cc/experimental/base/public/ |
| D | tensorhandle.h | 45 static TensorHandle FromTensor(const Tensor& tensor, const Runtime& runtime,
83 inline TensorHandle TensorHandle::FromTensor(const Tensor& tensor, in FromTensor() function
|
| /external/tensorflow/tensorflow/cc/experimental/base/tests/ |
| D | tensorhandle_test.cc | 69 TensorHandle::FromTensor(original_tensor, *runtime, &status); in TYPED_TEST()
111 TensorHandle::FromTensor(original_tensor, *runtime, &status); in TYPED_TEST()
161 TensorHandle::FromTensor(original_tensor, *runtime, &status); in TYPED_TEST()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_tensor.cc | 21 /*static*/ XlaTensor* XlaTensor::FromTensor(const Tensor* tensor) { in FromTensor() function in tensorflow::XlaTensor
32 const XlaTensor* xla_tensor = FromTensor(&tensor); in DeviceMemoryFromTensor()
|
| D | xla_tensor.h | 40 static XlaTensor* FromTensor(const Tensor* tensor);
|
| D | xla_tpu_device.cc | 75 tensorflow::XlaTensor::FromTensor(&tensor); in TpuPaddedShapeFn()
194 XlaTensor* const xla_input = XlaTensor::FromTensor(input); in TpuDeviceToDeviceCopy()
196 XlaTensor* const xla_output = XlaTensor::FromTensor(output); in TpuDeviceToDeviceCopy()
|
| D | xla_device_context.cc | 133 XlaTensor* xla_tensor = XlaTensor::FromTensor(device_tensor); in CopyCPUTensorToDevice()
239 XlaTensor* xla_tensor = XlaTensor::FromTensor(device_tensor); in CopyDeviceTensorToCPU()
|
| D | xla_launch_util.cc | 262 XlaTensor* xla_tensor = XlaTensor::FromTensor(t); in PopulateInputs()
276 XlaTensor* xla_tensor = XlaTensor::FromTensor(t); in PopulateInputs()
334 XlaTensor* xla_tensor = XlaTensor::FromTensor(output_tensor); in PopulateXlaTensor()
|
| D | xla_device.cc | 69 tensorflow::XlaTensor::FromTensor(&tensor); in DefaultPaddedShapeFn()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_execute_op.cc | 215 XlaTensor* xla_tensor = XlaTensor::FromTensor(&tensor); in BuildComputationInputs()
315 XlaTensor* xla_tensor = XlaTensor::FromTensor(&tensor); in BuildComputationInputs()
479 XlaTensor* xla_tensor = XlaTensor::FromTensor(output_tensor); in AllocateOutputTensors()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | converter_test.cc | 93 TEST(FromTensor, Smoke) { in TEST() argument
|