Searched defs:BroadcastTo (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | broadcast_to_op.h | 33 struct BroadcastTo { struct
35 void DoBCast( in DoBCast()
47 void ReshapeAndBCast(const Device &device, Tensor &output_tensor, in ReshapeAndBCast()
58 void operator()(const Device &device, OpKernelContext *ctx, in operator()
|
| H A D | broadcast_to_op_test.cc | 27 static Graph* BroadcastTo(int dim0, int dim1, InputShape input_shape) { in BroadcastTo() function
|
| H A D | matmul_op_test.cc | 474 Node* BroadcastTo(Graph* g, Node* input, Node* shape) { in BroadcastTo() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| H A D | broadcast.cc | 32 StatusOr<xla::XlaOp> BroadcastTo(xla::XlaOp input, in BroadcastTo() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | broadcast.cc | 28 StatusOr<XlaOp> BroadcastTo(XlaOp input, in BroadcastTo() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | broadcast_to.h | 62 inline void BroadcastTo(const RuntimeShape& unextended_input_shape, in BroadcastTo() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/ |
| H A D | matrix_triangular_solve_op_test.cc | 32 Node* BroadcastTo(Graph* g, Node* input, Node* shape) { in BroadcastTo() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | randomized_tests.cc | 2066 TEST_F(OpTest, BroadcastTo) { in TEST_F() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/go/op/ |
| H A D | wrappers.go | 4927 func BroadcastTo(scope *Scope, input tf.Output, shape tf.Output) (output tf.Output) { func
|