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()
|