Searched defs:BroadcastTo (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | broadcast_to_op.h | 33 struct BroadcastTo { struct
35 void DoBCast32Bit(const Device &device, typename TTypes<T, NDIMS>::Tensor out, in DoBCast32Bit()
42 void DoBCast( in DoBCast()
50 void ReshapeAndBCast(const Device &device, Tensor &output_tensor, in ReshapeAndBCast()
71 void operator()(const Device &device, OpKernelContext *ctx, in operator()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | broadcast.cc | 28 xla::StatusOr<xla::XlaOp> BroadcastTo(xla::XlaOp input, in BroadcastTo() function
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 2887 func BroadcastTo(scope *Scope, input tf.Output, shape tf.Output) (output tf.Output) { func
|