Searched refs:y_reshape (Results 1 – 12 of 12) 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/kernels/ |
| D | substr_op.cc | 146 auto pos_shaped = pos_tensor.shaped<T, 1>(bcast.y_reshape()); in Compute()
147 auto len_shaped = len_tensor.shaped<T, 1>(bcast.y_reshape()); in Compute()
209 auto pos_shaped = pos_tensor.shaped<T, 2>(bcast.y_reshape()); in Compute()
210 auto len_shaped = len_tensor.shaped<T, 2>(bcast.y_reshape()); in Compute()
|
| D | broadcast_to_op.h | 75 const int ndims = bcast.y_reshape().size(); in operator()
|
| D | cwise_ops_common.h | 128 in1.template shaped<Tin, 2>(bcast->y_reshape()), in Compute()
135 in1.template shaped<Tin, 3>(bcast->y_reshape()), in Compute()
142 in1.template shaped<Tin, 4>(bcast->y_reshape()), in Compute()
149 in1.template shaped<Tin, 5>(bcast->y_reshape()), in Compute()
|
| D | sparse_dense_binary_op_shared.cc | 125 dense_t->shaped<T, NDIM>(b.y_reshape()) \ in Compute()
|
| D | xent_op.cc | 93 labels_in.template shaped<T, 2>(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()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | bcast.h | 97 const Vec& y_reshape() const { return y_reshape_; } in y_reshape() function
|
| D | bcast_test.cc | 35 strings::StrAppend(&ret, "[", str_util::Join(b.y_reshape(), ","), "]"); in BCast()
|