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

  /external/tensorflow/tensorflow/compiler/xla/tests/
outfeed_in_nested_computation_test.cc 35 Shape xfeed_shape = ShapeUtil::MakeShape(xla::S32, {2}); local
58 XlaOp some_input = Infeed(&body_builder, xfeed_shape);
60 Outfeed(sum, xfeed_shape, "");
103 local_client_->TransferFromOutfeed(&xfeed_shape));
  /external/tensorflow/tensorflow/compiler/xla/tools/
replay_computation.cc 142 absl::optional<Shape> xfeed_shape; local
149 xfeed_shape = std::move(ParseShape(fake_xfeed_shape)).ValueOrDie();
161 xfeed_shape = is_infeed
167 << ShapeUtil::HumanString(*xfeed_shape);
184 return xfeed_shape;

Completed in 112 milliseconds