Search Options

Results per page
Sort
Preferred Languages
Advance

Results 101 - 110 of 234 for eraseOp (0.21 sec)

  1. tensorflow/compiler/aot/codegen_test.cc

      // Get rid of all CR characters, we may be running under windows.
      string sanitized_expected_contents(expected_contents);
      if (ignore_cr) {
        sanitized_expected_contents.erase(
            std::remove(sanitized_expected_contents.begin(),
                        sanitized_expected_contents.end(), '\r'),
            sanitized_expected_contents.end());
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 01 02:13:40 UTC 2024
    - 10.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/region_control_flow_to_functional.cc

          then_name, else_name, if_region.getIsStateless());
      CopyAndOverrideAttributes(if_region, if_op, &builder);
    
      if_region.replaceAllUsesWith(if_op.getResults());
      if_region.erase();
    
      if (to_bool && to_bool.use_empty()) to_bool.erase();
      return success();
    }
    
    // Transform CaseRegion to Case.
    LogicalResult RegionControlFlowToFunctional::ConvertCaseOp(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 28.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/xla_inline_device_ops.cc

      Value old_val, new_val;
      for (auto it : llvm::zip(cluster_op.getResults(), return_op.getOperands())) {
        std::tie(old_val, new_val) = it;
        old_val.replaceAllUsesExcept(new_val, &return_op);
      }
    
      cluster_op.erase();
    }
    
    void XlaInlineDeviceOpsPass::runOnOperation() {
      ModuleOp module = getOperation();
    
      llvm::SmallVector<tf_device::ClusterOp, 4> ops;
      module.walk(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 04 00:59:46 UTC 2022
    - 2.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tf2xla/transforms/tfxla_device_specific_transforms.cc

      OpBuilder opbuilder(get_alg_op);
    
      auto tf_const = opbuilder.create<TF::ConstOp>(
          get_alg_op->getLoc(), opbuilder.getI32IntegerAttr((int)tensorflow_rng));
    
      get_alg_op->replaceAllUsesWith(tf_const);
      get_alg_op->erase();
      return success();
    }
    
    void TFXLADeviceSpecificTransforms::runOnOperation() {
      if (!device_type_.hasValue()) return;
      auto func_op = getOperation();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 03 05:56:39 UTC 2023
    - 3.1K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/utils/xla_rewrite_util.h

    #include "xla/xla_data.pb.h"
    #include "tensorflow/core/lib/core/status.h"
    #include "tensorflow/core/platform/statusor.h"
    #include "tensorflow/core/util/device_name_utils.h"
    
    namespace tensorflow {
    // Erase rewritten ClusterFuncOp(s). If TPUPartitionedInputV2Op /
    // TPUPartitionedOutputV2Op are present, they must be removed along with the
    // ClusterFuncOp(s).
    mlir::LogicalResult EraseClusterFuncs(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 2.7K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/transforms/modify_io_nodes.cc

            return failure();
          }
          new_input_types[i] = arg_type;
          arg.dropAllUses();
          if (quantize_op.use_empty()) {
            quantize_op.erase();
          }
        } else {
          // `arg` has multiple uses or the user isn't a quantiz op (so we couldn't
          // rewrite it to a different type. Make a copy of the `arg` and replace
          // its use.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.9K bytes
    - Viewed (0)
  7. platforms/core-configuration/kotlin-dsl/src/test/kotlin/org/gradle/kotlin/dsl/execution/ProgramParserTest.kt

            assertEmptyProgram(" /* before */buildscript { /* within */ }/* after */ ")
        }
    
        @Test
        fun `empty Stage 1 with non-empty Stage 2 parse to Stage 2 with Stage 1 fragments erased`() {
    
            val scriptOnlySource =
                programSourceWith("println(\"Stage 2\")")
    
            assertProgramOf(
                scriptOnlySource,
                Program.Script(scriptOnlySource)
            )
    
    Registered: Wed Jun 12 18:38:38 UTC 2024
    - Last Modified: Wed Aug 02 08:06:49 UTC 2023
    - 8.1K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/transforms/convert_launch_func_to_tf_call.cc

            /*executor_type=*/builder.getStringAttr(""));
        call_op->setAttr("device", launch->getAttrOfType<StringAttr>("device"));
        launch.replaceAllUsesWith(call_op);
        launch.erase();
      });
    }
    
    }  // namespace
    
    std::unique_ptr<OperationPass<ModuleOp>> CreateConvertLaunchFuncToTFCallPass() {
      return std::make_unique<ConvertLaunchFuncToTFCallPass>();
    }
    
    }  // namespace TFDevice
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Aug 31 21:08:09 UTC 2023
    - 2.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/name_anonymous_iterators.cc

          op->getLoc(), op->getResultTypes()[0], name, /*container=*/"",
          op.getOutputTypes(), op.getOutputShapes());
      op->getResults()[0].replaceAllUsesWith(new_op->getResults()[0]);
      if (op->use_empty()) op->erase();
      return count;
    }
    
    void NameAnonymousIteratorsPass::runOnOperation() {
      int count = 1;
      getOperation().walk(
          [&](TF::AnonymousIteratorOp op) { count = replace(op, count); });
      getOperation().walk(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Nov 03 12:35:38 UTC 2022
    - 2.7K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tfr/passes/rewrite_quantized_io.cc

            auto new_op = builder.create<TFR::CastOp>(
                returned_op->getLoc(), new_type, returned_op.getArg());
            returned_value.set(new_op.getResult());
            if (returned_op.use_empty()) {
              returned_op.erase();
            }
          } else {
            returned_value.get().getDefiningOp()->emitError(
                "The producer of quantized type result should be a tfr.cast op.");
            return;
          }
        }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 08 01:19:25 UTC 2023
    - 4.5K bytes
    - Viewed (0)
Back to top