Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/grappler/optimizers/
Devaluation_utils_test.cc44 EXPECT_FALSE(dsimple.MakeTensorFromProto(proto, {}, &tensor).ok()); in TEST()
50 TF_ASSERT_OK(dsimple.MakeTensorFromProto(proto, {}, &tensor)); in TEST()
Devaluation_utils.h40 Status MakeTensorFromProto(const TensorProto& tensor_proto,
Devaluation_utils.cc71 Status DeviceSimple::MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto() function in tensorflow::grappler::DeviceSimple
/external/tensorflow/tensorflow/core/common_runtime/
Drenamed_device.h91 Status MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto() function
94 return underlying_->MakeTensorFromProto(tensor_proto, alloc_attrs, tensor); in MakeTensorFromProto()
Dthreadpool_device.h38 Status MakeTensorFromProto(const TensorProto& tensor_proto,
Dsingle_threaded_cpu_device.cc65 Status MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto() function in tensorflow::__anona485c93d0111::SingleThreadedCpuDevice
Dthreadpool_device.cc91 Status ThreadPoolDevice::MakeTensorFromProto( in MakeTensorFromProto() function in tensorflow::ThreadPoolDevice
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compilation_device.h59 Status MakeTensorFromProto(const TensorProto& tensor_proto,
Dxla_compilation_device.cc118 Status XlaCompilationDevice::MakeTensorFromProto( in MakeTensorFromProto() function in tensorflow::XlaCompilationDevice
/external/tensorflow/tensorflow/core/distributed_runtime/
Dtensor_coding.cc61 s = device_->MakeTensorFromProto(meta_.tensor(), alloc_attrs_, &tensor_); in InitFrom()
92 device_->MakeTensorFromProto(meta_.tensor(), alloc_attrs_, &tensor_); in ParseFrom()
/external/tensorflow/tensorflow/core/kernels/
Dhost_constant_op.cc34 ctx, ctx->device()->MakeTensorFromProto(*proto, alloc_attr, &tensor_)); in _HostConstantOp()
Dparse_tensor_op.cc51 OP_REQUIRES_OK(ctx, ctx->device()->MakeTensorFromProto( in Compute()
Dsummary_image_op.cc41 OP_REQUIRES_OK(context, context->device()->MakeTensorFromProto( in SummaryImageOp()
Dremote_fused_graph_execute_utils.h131 static Status MakeTensorFromProto(const TensorProto& tensor_proto,
Dremote_fused_graph_execute_op_test.cc189 TF_CHECK_OK(RemoteFusedGraphExecuteUtils::MakeTensorFromProto( in ExecuteGraph()
Dconstant_op.cc75 OP_REQUIRES_OK(ctx, ctx->device()->MakeTensorFromProto( in ConstantOp()
/external/tensorflow/tensorflow/core/common_runtime/sycl/
Dsycl_device.cc45 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/core/framework/
Ddevice_base.h243 virtual Status MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto() function
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.h151 Status MakeTensorFromProto(const TensorProto& tensor_proto,
Dxla_device.cc440 Status XlaDevice::MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto() function in tensorflow::XlaDevice
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transferer.h209 static Status MakeTensorFromProto(const TensorProto& tensor_proto,
Dgraph_transferer.cc496 TF_CHECK_OK(MakeTensorFromProto(*proto, &const_tensor)); in RegisterConstantNode()
741 TF_CHECK_OK(MakeTensorFromProto(*tensor_proto, &const_tensor)); in RegisterPadNode()
751 TF_CHECK_OK(MakeTensorFromProto(*proto, &const_tensor)); in RegisterPadNode()
1100 /* static */ Status GraphTransferer::MakeTensorFromProto( in MakeTensorFromProto() function in tensorflow::GraphTransferer
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.h89 Status MakeTensorFromProto(const TensorProto& tensor_proto,
/external/tensorflow/tensorflow/contrib/mpi/
Dmpi_rendezvous_mgr.cc120 dst_device->MakeTensorFromProto(mpi_response.response().tensor(), in RecvFromRemoteAsync()

12