Searched defs:BroadcastInDim (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ | ||
| D | local_computation_builder.cc | 475 LocalOp ComputationBuilder::BroadcastInDim( in BroadcastInDim() function in xla::swig::ComputationBuilder |
| D | xla_client.py | 1104 def BroadcastInDim(self, operand, shape, broadcast_dimensions): member in ComputationBuilder |
| /external/tensorflow/tensorflow/compiler/xla/client/ | ||
| D | xla_builder_test.cc | 302 TEST_F(XlaBuilderTest, BroadcastInDim) { in TEST_F() argument |
| D | xla_builder.cc | 671 XlaOp XlaBuilder::BroadcastInDim( in BroadcastInDim() function in xla::XlaBuilder |