- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for set_shape (0.14 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
// [H, W, ..., filter_in_depth, G, out_depth / G]. auto new_shape = llvm::to_vector<6>(filter_shape); new_shape.back() = feature_group_count; new_shape.push_back(filter_shape.back() / feature_group_count); Type filter_element_ty = filter_ty.getElementType(); auto ty = tensorflow::GetTypeFromTFTensorShape(new_shape, filter_element_ty);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0)