Searched refs:FromShape (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | betainc_op.cc | 83 auto merged_shape_vec = BCast::FromShape(merged_shape); in Compute()
84 BCast a_shaper(BCast::FromShape(a_shape), merged_shape_vec); in Compute()
85 BCast b_shaper(BCast::FromShape(b_shape), merged_shape_vec); in Compute()
86 BCast x_shaper(BCast::FromShape(x_shape), merged_shape_vec); in Compute()
|
| D | cwise_ops_common.cc | 58 bcast(BCast::FromShape(in0.shape()), BCast::FromShape(in1.shape())) { in BinaryOpState()
|
| D | broadcast_to_op.cc | 77 BCast bcast(BCast::FromShape(input_shape), BCast::FromShape(output_shape), in Compute()
|
| D | sparse_dense_binary_op_shared.cc | 84 const auto lhs_dims = BCast::FromShape(TensorShape(shape_vec)); in Compute()
85 const auto rhs_dims = BCast::FromShape(dense_t->shape()); in Compute()
|
| D | xent_op.cc | 49 BCast bcast(BCast::FromShape(logits_in.shape()), in Compute()
50 BCast::FromShape(labels_in.shape())); in Compute()
|
| D | substr_op.cc | 131 BCast bcast(BCast::FromShape(input_shape), BCast::FromShape(pos_shape)); in Compute()
|
| D | quantized_mul_op.cc | 292 BCast bcast(BCast::FromShape(x.shape()), BCast::FromShape(y.shape())); in Compute()
|
| D | quantized_add_op.cc | 465 BCast bcast(BCast::FromShape(x.shape()), BCast::FromShape(y.shape())); in Compute()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | cwise_ops.cc | 43 BCast bcast(BCast::FromShape(lhs_shape), BCast::FromShape(rhs_shape), in Compile()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | bcast.h | 105 static Vec FromShape(const TensorShape& shape);
|
| D | bcast.cc | 158 BCast::Vec BCast::FromShape(const TensorShape& shape) { in FromShape() function in tensorflow::BCast
|