Home
last modified time | relevance | path

Searched refs:MakeTensorFromProto (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Drenamed_device.h85 Status MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto() function
88 return underlying_->MakeTensorFromProto(tensor_proto, alloc_attrs, tensor); in MakeTensorFromProto()
Dthreadpool_device.h34 Status MakeTensorFromProto(const TensorProto& tensor_proto,
Dthreadpool_device.cc68 Status ThreadPoolDevice::MakeTensorFromProto( in MakeTensorFromProto() function in tensorflow::ThreadPoolDevice
/external/tensorflow/tensorflow/core/distributed_runtime/
Dtensor_coding.cc61 s = device_->MakeTensorFromProto(meta_.tensor(), alloc_attrs_, &tensor_); in InitFrom()
91 device_->MakeTensorFromProto(meta_.tensor(), alloc_attrs_, &tensor_); in ParseFrom()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compilation_device.h62 Status MakeTensorFromProto(const TensorProto& tensor_proto,
Dxla_compilation_device.cc120 Status XlaCompilationDevice::MakeTensorFromProto( in MakeTensorFromProto() function in tensorflow::XlaCompilationDevice
/external/tensorflow/tensorflow/core/kernels/
Dparse_tensor_op.cc51 OP_REQUIRES_OK(ctx, ctx->device()->MakeTensorFromProto( in Compute()
Dconstant_op.cc74 OP_REQUIRES_OK(ctx, ctx->device()->MakeTensorFromProto( in ConstantOp()
139 ctx, ctx->device()->MakeTensorFromProto(*proto, alloc_attr, &tensor_)); in HostConstantOp()
Dsummary_image_op.cc41 OP_REQUIRES_OK(context, context->device()->MakeTensorFromProto( in SummaryImageOp()
Dremote_fused_graph_execute_op_test.cc189 TF_CHECK_OK(RemoteFusedGraphExecuteUtils::MakeTensorFromProto( in ExecuteGraph()
Dremote_fused_graph_execute_utils.h122 static Status MakeTensorFromProto(const TensorProto& tensor_proto,
Dremote_fused_graph_execute_utils.cc352 /* static */ Status RemoteFusedGraphExecuteUtils::MakeTensorFromProto( in MakeTensorFromProto() function in tensorflow::RemoteFusedGraphExecuteUtils
/external/tensorflow/tensorflow/core/common_runtime/sycl/
Dsycl_device.cc46 Status SYCLDevice::MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto() function in tensorflow::SYCLDevice
Dsycl_device.h215 Status MakeTensorFromProto(const TensorProto& tensor_proto,
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.h94 Status MakeTensorFromProto(const TensorProto& tensor_proto,
Dxla_device.cc258 Status XlaDevice::MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto() function in tensorflow::XlaDevice
/external/tensorflow/tensorflow/core/framework/
Ddevice_base.h218 virtual Status MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto() function
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transferer.h206 static Status MakeTensorFromProto(const TensorProto& tensor_proto,
Dgraph_transferer.cc489 TF_CHECK_OK(MakeTensorFromProto(*proto, &const_tensor)); in RegisterConstantNode()
734 TF_CHECK_OK(MakeTensorFromProto(*tensor_proto, &const_tensor)); in RegisterPadNode()
744 TF_CHECK_OK(MakeTensorFromProto(*proto, &const_tensor)); in RegisterPadNode()
1097 /* static */ Status GraphTransferer::MakeTensorFromProto( in MakeTensorFromProto() function in tensorflow::GraphTransferer
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.h80 Status MakeTensorFromProto(const TensorProto& tensor_proto,
Dgpu_device.cc582 Status BaseGPUDevice::MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto() function in tensorflow::BaseGPUDevice
/external/tensorflow/tensorflow/contrib/mpi/
Dmpi_rendezvous_mgr.cc120 dst_device->MakeTensorFromProto(mpi_response.response().tensor(), in RecvFromRemoteAsync()
/external/tensorflow/tensorflow/contrib/verbs/
Drdma.cc1650 Status s = dst_dev_->MakeTensorFromProto(proto, recv_args_.alloc_attrs, in RecvTensorContent()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc81 Status MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto() function in tensorflow::grappler::__anon9a0b719b0111::DeviceSimple