Home
last modified time | relevance | path

Searched refs:representation_shape (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_kernel.cc390 TF_ASSIGN_OR_RETURN(xla::Shape representation_shape, in ReadVariableInputTensor()
396 if (xla::ShapeUtil::Compatible(xla_shape, representation_shape)) { in ReadVariableInputTensor()
529 xla::Shape representation_shape, in AssignVariableTensor()
533 if (!xla::ShapeUtil::Compatible(xla_shape, representation_shape)) { in AssignVariableTensor()
535 xla::AsInt64Slice(representation_shape.dimensions())); in AssignVariableTensor()
537 variable->SetRepresentationShape(representation_shape); in AssignVariableTensor()
Dxla_resource.h91 const absl::optional<xla::Shape>& representation_shape() const { in representation_shape() function
Dxla_compiler.cc294 if (resource->representation_shape().has_value()) { in BuildComputation()
296 elems.size(), resource->representation_shape()->layout()); in BuildComputation()