Searched refs:FromTensor (Results 1 – 13 of 13) 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 | 23 /*static*/ XlaTensor* XlaTensor::FromTensor(const Tensor* tensor) { in FromTensor() function in tensorflow::XlaTensor
34 const XlaTensor* xla_tensor = FromTensor(&tensor); in DeviceMemoryFromTensor()
|
| D | xla_tensor.h | 40 static XlaTensor* FromTensor(const Tensor* tensor);
|
| D | xla_tpu_device.cc | 78 tensorflow::XlaTensor::FromTensor(&tensor); in TpuPaddedShapeFn()
197 XlaTensor* const xla_input = XlaTensor::FromTensor(input); in TpuDeviceToDeviceCopy()
199 XlaTensor* const xla_output = XlaTensor::FromTensor(output); in TpuDeviceToDeviceCopy()
|
| D | xla_device_context.cc | 133 XlaTensor* xla_tensor = XlaTensor::FromTensor(device_tensor); in CopyCPUTensorToDevice()
241 XlaTensor* xla_tensor = XlaTensor::FromTensor(device_tensor); in CopyDeviceTensorToCPU()
|
| D | xla_launch_util.cc | 290 XlaTensor* xla_tensor = XlaTensor::FromTensor(t); in PopulateInputs()
360 XlaTensor* xla_tensor = XlaTensor::FromTensor(&output_tensor); in GetOrCreateTensorForOutput()
|
| D | xla_device.cc | 69 tensorflow::XlaTensor::FromTensor(&tensor); in DefaultPaddedShapeFn()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_reshard_variables_op_util.cc | 117 XlaTensor* xla_tensor = XlaTensor::FromTensor(&tensor); in BuildInputBuffers()
182 XlaTensor* xla_tensor = XlaTensor::FromTensor(&tensor); in BuildInputBuffers()
293 XlaTensor* xla_tensor = XlaTensor::FromTensor(output_tensor); in UpdateOutputVariables()
|
| D | tpu_execute_op.cc | 214 XlaTensor* xla_tensor = XlaTensor::FromTensor(&tensor); in BuildComputationInputs()
314 XlaTensor* xla_tensor = XlaTensor::FromTensor(&tensor); in BuildComputationInputs()
478 XlaTensor* xla_tensor = XlaTensor::FromTensor(output_tensor); in AllocateOutputTensors()
|
| /external/tensorflow/tensorflow/core/tfrt/common/ |
| D | async_value_tensor.h | 32 static AsyncValueTensor* FromTensor(const Tensor* tensor);
|
| D | async_value_tensor.cc | 27 /*static*/ AsyncValueTensor* AsyncValueTensor::FromTensor( in FromTensor() function in tensorflow::AsyncValueTensor
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | converter_test.cc | 93 TEST(FromTensor, Smoke) { in TEST() argument
|