Home
last modified time | relevance | path

Searched refs:ReshapeWithCorrectRepresentationAndSharding (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_helpers.cc180 xla::StatusOr<xla::XlaOp> ReshapeWithCorrectRepresentationAndSharding( in ReshapeWithCorrectRepresentationAndSharding() function
189 ReshapeWithCorrectRepresentationAndSharding( in ReshapeWithCorrectRepresentationAndSharding()
Dxla_helpers.h93 xla::StatusOr<xla::XlaOp> ReshapeWithCorrectRepresentationAndSharding(
Dxla_compiler.cc234 ReshapeWithCorrectRepresentationAndSharding( in BuildComputation()
333 handle, ReshapeWithCorrectRepresentationAndSharding( in BuildComputation()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc1321 tensorflow::ReshapeWithCorrectRepresentationAndSharding( in Lower()