| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | outfeed_in_nested_computation_test.cc | 46 Outfeed(loop_counter, int_shape, ""); in XLA_TEST_F()
60 Outfeed(sum, xfeed_shape, ""); in XLA_TEST_F()
130 Outfeed(param, result_shape, ""); in XLA_TEST_F()
|
| D | local_client_execute_test.cc | 878 Outfeed(sum, shape, /*outfeed_config=*/""); in XLA_TEST_F()
|
| D | BUILD | 1977 # Outfeed ops are not supported on the interpreter backend.
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_invariant_code_motion_test.cc | 294 Contains(op::Outfeed())); in TEST_F()
348 Contains(op::Outfeed())); in TEST_F()
|
| D | hlo_matchers.h | 220 HLO_MATCHER(Outfeed);
|
| D | hlo.proto | 108 // Outfeed configuration information, only present for kOutfeed.
|
| D | pattern_matcher.h | 2171 XLA_BINOP_PATTERN(Outfeed)
|
| /external/tensorflow/tensorflow/compiler/xla/rpc/ |
| D | xla_service.proto | 102 // Transferred literal from the Outfeed buffer of the device.
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 452 void ComputationBuilder::Outfeed(const LocalOp& operand, const Shape& shape, in Outfeed() function in xla::swig::ComputationBuilder
454 xla::Outfeed(operand.op(), shape, outfeed_config); in Outfeed()
|
| D | local_computation_builder.h | 231 void Outfeed(const LocalOp& operand, const Shape& shape,
|
| D | local_computation_builder.i | 295 %unignore xla::swig::ComputationBuilder::Outfeed;
|
| D | xla_client.py | 953 def Outfeed(self, operand): member in ComputationBuilder
959 self._client.Outfeed(operand, self.GetShape(operand), ''.encode('utf-8'))
|
| D | xla_client_test.py | 1665 c.Outfeed(x)
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 426 void Outfeed(const XlaOp& operand, const Shape& shape_with_layout,
820 friend void Outfeed(const XlaOp& operand, const Shape& shape_with_layout,
1409 void Outfeed(const XlaOp& operand, const Shape& shape_with_layout,
|
| D | xla_builder.cc | 1385 void XlaBuilder::Outfeed(const XlaOp& operand, const Shape& shape_with_layout, in Outfeed() function in xla::XlaBuilder
3071 void Outfeed(const XlaOp& operand, const Shape& shape_with_layout, in Outfeed() function
3073 return operand.builder()->Outfeed(operand, shape_with_layout, outfeed_config); in Outfeed()
|