Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_computation_builder.h189 class LocalOp {
191 LocalOp(const XlaOp& op);
217 StatusOr<Computation*> BuildWithRoot(const LocalOp& root);
219 LocalOp Parameter(int64 parameter_number, const Shape& shape,
222 StatusOr<Shape> GetShape(const LocalOp& operand);
227 LocalOp ReplicaId();
229 LocalOp Infeed(const Shape& shape);
231 void Outfeed(const LocalOp& operand, const Shape& shape,
234 LocalOp ConstantLiteral(const Literal& literal);
236 LocalOp Iota(PrimitiveType element_type, int64 size);
[all …]
Dlocal_computation_builder.cc409 LocalOp::LocalOp(const XlaOp& op) : op_(op) {} in LocalOp() function in xla::swig::LocalOp
411 const XlaOp& LocalOp::op() const { return op_; } in op()
427 LocalOp ComputationBuilder::Parameter(int64 parameter_number, in Parameter()
432 StatusOr<Computation*> ComputationBuilder::BuildWithRoot(const LocalOp& root) { in BuildWithRoot()
437 StatusOr<Shape> ComputationBuilder::GetShape(const LocalOp& operand) { in GetShape()
446 LocalOp ComputationBuilder::ReplicaId() { return xla::ReplicaId(&builder_); } in ReplicaId()
448 LocalOp ComputationBuilder::Infeed(const Shape& shape) { in Infeed()
452 void ComputationBuilder::Outfeed(const LocalOp& operand, const Shape& shape, in Outfeed()
457 LocalOp ComputationBuilder::ConstantLiteral(const Literal& literal) { in ConstantLiteral()
461 LocalOp ComputationBuilder::Iota(PrimitiveType element_type, int64 size) { in Iota()
[all …]
Dlocal_computation_builder.i53 %typemap(in) absl::Span<const xla::swig::LocalOp>(
54 std::vector<LocalOp> temps) {
62 LocalOp* op;
63 if ((SWIG_ConvertPtr(o, (void**)&op, $descriptor(xla::swig::LocalOp*),
283 %unignore xla::swig::LocalOp;