Searched refs:const_vec_1 (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/ |
| D | reshape_vectorizer.cc | 35 Output const_vec_1 = ops::Const(*s, {1}); in GetVectorizedShape() local
40 ops::StridedSlice(*s, shape, const_vec_1, const_vec_1, const_vec_1, in GetVectorizedShape()
|
| D | cwise_op_vectorizer.cc | 88 Output const_vec_1 = ops::Const(scope, {1}); in ExpandDimsForBroadcast() local
91 scope, shape, const_vec_1, const_vec_1, const_vec_1, in ExpandDimsForBroadcast()
96 scope, shape, const_vec_1, const_vec_1, const_vec_1, in ExpandDimsForBroadcast()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | vectorization_utils.cc | 453 Node* const_vec_1; in StackTensor() local
454 TF_RETURN_IF_ERROR(make_const({1}, g, &const_vec_1)); in StackTensor()
460 .Input(const_vec_1) // end in StackTensor()
461 .Input(const_vec_1) // strides in StackTensor()
467 .Input(const_vec_1) // shape in StackTensor()
|