Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/grappler/optimizers/
Devaluation_utils_test.cc42 EXPECT_FALSE(dsimple.MakeTensorFromProto(proto, {}, &tensor).ok()); in TEST()
48 TF_ASSERT_OK(dsimple.MakeTensorFromProto(proto, {}, &tensor)); in TEST()
Devaluation_utils.h40 Status MakeTensorFromProto(const TensorProto& tensor_proto,
Devaluation_utils.cc50 Status DeviceSimple::MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto() function in tensorflow::grappler::DeviceSimple
/external/tensorflow/tensorflow/core/common_runtime/
Drenamed_device.h106 Status MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto() function
109 return underlying_device_->MakeTensorFromProto(tensor_proto, alloc_attrs, in MakeTensorFromProto()
Dthreadpool_device.h38 Status MakeTensorFromProto(const TensorProto& tensor_proto,
Dsingle_threaded_cpu_device.cc61 Status MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto() function in tensorflow::__anonfe4177f40111::SingleThreadedCpuDevice
Dthreadpool_device.cc88 Status ThreadPoolDevice::MakeTensorFromProto( in MakeTensorFromProto() function in tensorflow::ThreadPoolDevice
/external/tensorflow/tensorflow/core/tpu/
Dvirtual_device.h31 Status MakeTensorFromProto(const TensorProto& tensor_proto,
Dvirtual_device.cc79 Status VirtualDevice::MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto() function in tensorflow::VirtualDevice
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compilation_device.h59 Status MakeTensorFromProto(const TensorProto& tensor_proto,
Dxla_compilation_device.cc146 Status XlaCompilationDevice::MakeTensorFromProto( in MakeTensorFromProto() function in tensorflow::XlaCompilationDevice
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.h155 Status MakeTensorFromProto(const TensorProto& tensor_proto,
205 Status MakeTensorFromProto(XlaDeviceContext* device_context,
Dxla_device.cc474 Status XlaDevice::MakeTensorFromProto(XlaDeviceContext* device_context, in MakeTensorFromProto() function in tensorflow::XlaDevice
499 Status XlaDevice::MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto() function in tensorflow::XlaDevice
508 return MakeTensorFromProto(device_contexts.first, tensor_proto, alloc_attrs, in MakeTensorFromProto()
521 return MakeTensorFromProto(device_contexts.second, tensor_proto, alloc_attrs, in MakeFastMemTensorFromProto()
/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.cc78 OP_REQUIRES_OK(ctx, ctx->device()->MakeTensorFromProto( in ConstantOp()
Dremote_fused_graph_execute_utils.cc354 /* static */ Status RemoteFusedGraphExecuteUtils::MakeTensorFromProto( in MakeTensorFromProto() function in tensorflow::RemoteFusedGraphExecuteUtils
/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/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/framework/
Ddevice_base.h238 virtual Status MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto() function
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.h72 Status MakeTensorFromProto(const TensorProto& tensor_proto,

12