Searched defs:trans_dims (Results 1 – 1 of 1) sorted by relevance
774 std::vector<int64_t> trans_dims(new_dimensions.size()); in SplitAndTransposeMergedBatch() local2583 std::vector<int64_t> trans_dims = retval.transpose_dims; in PropagateOnConv() local2823 std::vector<int64_t> trans_dims(activations->shape().dimensions_size()); in TransposeAndMergeBatch() local3451 std::vector<int64_t> trans_dims(convolution->shape().dimensions_size()); in PropagateOnBackpropFilterConv() local