Home
last modified time | relevance | path

Searched refs:SetTypeAndShape (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_resource.h100 Status SetTypeAndShape(DataType type, const TensorShape& shape);
Dxla_resource.cc89 Status XlaResource::SetTypeAndShape(DataType type, const TensorShape& shape) { in SetTypeAndShape() function in tensorflow::XlaResource
Dxla_op_kernel.cc526 TF_RETURN_IF_ERROR(variable->SetTypeAndShape(type, shape)); in AssignVariableTensor()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstack_ops.cc76 TF_RETURN_IF_ERROR(resource->SetTypeAndShape(dtype, elem_shape)); in MaybeInitializeStack()
Dtensor_array_ops.cc67 TF_RETURN_IF_ERROR(resource->SetTypeAndShape(dtype, elem_shape)); in MaybeInitializeTensorArray()
Dwhile_op.cc361 resource->SetTypeAndShape(update.type, update.shape)); in Compile()