Searched refs:y_reshape (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | test_graph_tfadd.pbtxt | 34 name : "y_reshape"
46 input : "y_reshape"
|
| D | test_graph_tfadd.config.pbtxt | 9 id { node_name: "y_reshape" }
|
| /external/tensorflow/tensorflow/core/util/ |
| D | matmul_bcast.h | 50 y_batch_size_ = TensorShape(batch_bcast_->y_reshape()).num_elements(); in MatMulBCast()
59 ComputeBatchIndices(output_batch_size_, batch_bcast_->y_reshape(), in MatMulBCast()
|
| D | bcast_test.cc | 35 strings::StrAppend(&ret, "[", absl::StrJoin(b.y_reshape(), ","), "]"); in BCast()
|
| D | bcast.h | 376 const Vec& y_reshape() const { return reshape_[1]; } in y_reshape() function
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | substr_op.cc | 151 auto pos_shaped = pos_tensor.shaped<T, 1>(bcast.y_reshape()); in Compute()
152 auto len_shaped = len_tensor.shaped<T, 1>(bcast.y_reshape()); in Compute()
204 auto pos_shaped = pos_tensor.shaped<T, 2>(bcast.y_reshape()); in Compute()
205 auto len_shaped = len_tensor.shaped<T, 2>(bcast.y_reshape()); in Compute()
|
| D | cwise_op_select.cc | 199 cond->template shaped<bool, NDIMS>(cond_bcast.y_reshape()), \ in Compute()
200 then->template shaped<T, NDIMS>(then_bcast.y_reshape()), \ in Compute()
201 else_->template shaped<T, NDIMS>(else_bcast.y_reshape()), \ in Compute()
|
| D | broadcast_to_op.h | 75 const int ndims = bcast.y_reshape().size(); in operator()
|
| D | xent_op.cc | 90 labels_in.template shaped<T, 2>(bcast.y_reshape()), in Compute()
|
| D | sparse_dense_binary_op_shared.cc | 125 dense_t->shaped<T, NDIM>(b.y_reshape()) \ in Compute()
|
| D | cwise_ops_common.h | 189 in1.template shaped<Tin, 2>(bcast.y_reshape()), in Compute()
196 in1.template shaped<Tin, 3>(bcast.y_reshape()), in Compute()
203 in1.template shaped<Tin, 4>(bcast.y_reshape()), in Compute()
210 in1.template shaped<Tin, 5>(bcast.y_reshape()), in Compute()
|
| D | betainc_op.cc | 101 output->shaped<T, NDIM>(a_shaper.y_reshape())); \ in Compute()
|
| D | quantized_mul_op.cc | 358 << TensorShape(bcast.y_reshape()).DebugString(); in Compute()
|
| D | quantized_add_op.cc | 550 << TensorShape(bcast.y_reshape()).DebugString(); in Compute()
|