Searched defs:BroadcastTo (Results 1 – 7 of 7) 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()
|
| D | broadcast_to_op_test.cc | 27 static Graph* BroadcastTo(int dim0, int dim1, InputShape input_shape) { in BroadcastTo() function
|
| D | matmul_op_test.cc | 467 Node* BroadcastTo(Graph* g, Node* input, Node* shape) { in BroadcastTo() function
|
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | broadcast.cc | 31 xla::StatusOr<xla::XlaOp> BroadcastTo(xla::XlaOp input, in BroadcastTo() function
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | broadcast_to.h | 62 inline void BroadcastTo(const RuntimeShape& unextended_input_shape, in BroadcastTo() function
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_triangular_solve_op_test.cc | 32 Node* BroadcastTo(Graph* g, Node* input, Node* shape) { in BroadcastTo() function
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 51045 func BroadcastTo(scope *Scope, input tf.Output, shape tf.Output) (output tf.Output) { func
|