Searched refs:trans_dims (Results 1 – 1 of 1) sorted by relevance
774 std::vector<int64_t> trans_dims(new_dimensions.size()); in SplitAndTransposeMergedBatch() local775 absl::c_iota(trans_dims, 0); in SplitAndTransposeMergedBatch()2583 std::vector<int64_t> trans_dims = retval.transpose_dims; in PropagateOnConv() local2584 CHECK(!trans_dims.empty()); in PropagateOnConv()2823 std::vector<int64_t> trans_dims(activations->shape().dimensions_size()); in TransposeAndMergeBatch() local2824 absl::c_iota(trans_dims, 0); in TransposeAndMergeBatch()2827 trans_dims[start_batch_dim_position + i] = in TransposeAndMergeBatch()2829 trans_dims[start_space_dim_position + i] = in TransposeAndMergeBatch()2833 TF_ASSIGN_OR_RETURN(activations, MakeTransposeHlo(activations, trans_dims)); in TransposeAndMergeBatch()3451 std::vector<int64_t> trans_dims(convolution->shape().dimensions_size()); in PropagateOnBackpropFilterConv() local[all …]