Searched refs:FromShape (Results 1 – 15 of 15) 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 | 79 BCast bcast(BCast::FromShape(input_shape), BCast::FromShape(output_shape), in Compute()
|
| D | xent_op.cc | 46 BCast bcast(BCast::FromShape(logits_in.shape()), in Compute()
47 BCast::FromShape(labels_in.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 | substr_op.cc | 136 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 | ragged_tensor_to_tensor_op.cc | 467 BCast bcast(BCast::FromShape(src_shape), BCast::FromShape(element_shape), in SetOutput()
|
| 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 | select_op.cc | 95 BCast bcast_then_else(BCast::FromShape(then_shape), in Compile()
96 BCast::FromShape(else_shape), in Compile()
104 BCast bcast(bcast_then_else.output_shape(), BCast::FromShape(cond_shape), in Compile()
|
| D | cwise_ops.cc | 43 BCast bcast(BCast::FromShape(lhs_shape), BCast::FromShape(rhs_shape), in Compile()
|
| D | matrix_triangular_solve_op.cc | 45 MatMulBCast bcast(BCast::FromShape(lhs_shape), BCast::FromShape(rhs_shape)); in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | broadcast.cc | 102 BCast bcast(BCast::FromShape(lhs_tf_shape), BCast::FromShape(rhs_tf_shape)); in BroadcastOpsToSame()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | bcast.cc | 21 BCast::Vec BCast::FromShape(const TensorShape& shape) { in FromShape() function in tensorflow::BCast
|
| D | bcast.h | 416 static Vec FromShape(const TensorShape& shape);
|