Searched refs:BroadcastTo (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | broadcast_to_op.cc | 84 functor::BroadcastTo<Device, T>()(device, ctx, *output_tensor, output_shape, in Compute()
104 void BroadcastTo<GPUDevice, Type>::operator()( \
108 extern template struct BroadcastTo<GPUDevice, Type>;
|
| D | broadcast_to_op_gpu.cu.cc | 28 template class functor::BroadcastTo<GPUDevice, Type>;
|
| D | broadcast_to_op.h | 33 struct BroadcastTo { struct
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | cwise_ops.cc | 92 auto lhs_output = BroadcastTo(lhs, broadcast_helper.output_shape()); in Broadcast()
97 auto rhs_output = BroadcastTo(rhs, broadcast_helper.output_shape()); in Broadcast()
|
| D | broadcast_to_op.cc | 35 auto output = BroadcastTo(context->Input(0), output_shape.dim_sizes()); in Compile()
|
| D | tile_ops.cc | 105 auto result = BroadcastTo(ctx->Input("input"), output_dims); in Compile()
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_BroadcastTo.pbtxt | 2 graph_op_name: "BroadcastTo"
|
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | broadcast.h | 27 xla::StatusOr<xla::XlaOp> BroadcastTo(xla::XlaOp input,
|
| D | broadcast.cc | 28 xla::StatusOr<xla::XlaOp> BroadcastTo(xla::XlaOp input, in BroadcastTo() function
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BroadcastTo.pbtxt | 2 graph_op_name: "BroadcastTo"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 516 name: "BroadcastTo"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 516 name: "BroadcastTo"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 2887 func BroadcastTo(scope *Scope, input tf.Output, shape tf.Output) (output tf.Output) { func
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 4981 name: "BroadcastTo"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 12195 name: "BroadcastTo"
|
| D | ops_history.v2.pbtxt | 12195 name: "BroadcastTo"
|