- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for shape_of (0.14 sec)
-
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
// CHECK-DAG: %[[SHAPEOF:.+]] = shape.shape_of %arg0 // CHECK-DAG: %[[CST0:.+]] = arith.constant 0 // CHECK-DAG: %[[CST1:.+]] = arith.constant 1 // CHECK-DAG: %[[GETEXTENT0:.+]] = tensor.extract %[[SHAPEOF]][%[[CST0]]] // CHECK-DAG: %[[CST1_0:.+]] = arith.constant 1 // CHECK-DAG: %[[GETEXTENT1:.+]] = tensor.extract %[[SHAPEOF]][%[[CST1_0]]]
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 335.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
auto element_type = x.getElementType(); auto shape_x = x.getShape(); auto shape_y = y.getShape(); if (shape_x.size() == shape_y.size()) { llvm::SmallVector<int64_t, 4> out_shape(shape_x.size()); for (int i = 0; i < shape_x.size(); i++) { auto x_val = shape_x[i]; auto y_val = shape_y[i]; out_shape[i] = std::max(x_val, y_val); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0)