HomeSort by relevance Sort by last modified time
    Searched refs:representation_shape (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_op_kernel.cc 310 TensorShape representation_shape = xla_context.VariableRepresentationShape( local
312 if (representation_shape == variable->shape()) {
433 TensorShape representation_shape = local
435 if (shape != representation_shape) {
436 handle = builder()->Reshape(handle, representation_shape.dim_sizes());
xla_compiler.cc 245 TensorShape representation_shape = local
247 return TensorShapeToXLAShape(arg.type, representation_shape,

Completed in 125 milliseconds