Search Options

Results per page
Sort
Preferred Languages
Advance

Results 31 - 40 of 90 for eraseOp (0.34 sec)

  1. tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc

          attr->contains("Truncate") && !attr->at("Truncate").b()) {
        node_def->set_op("Identity");
        attr->insert({{"T", attr->at("SrcT")}});
        attr->erase("SrcT");
        attr->erase("DstT");
        attr->erase("Truncate");
      }
    }
    
    }  // namespace
    
    Status GetAttrValuesFromOperation(
        mlir::Operation* inst, llvm::StringRef name,
        const tensorflow::OpRegistrationData* op_reg_data,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 11.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/merge_control_flow.cc

          /*operands=*/merged_else_yield_values);
    }
    
    // Merge the IfRegions in each segment. In the meantime, the old IfRegions in
    // the segment will be added to `regions_to_remove`. They will be erased in the
    // end.
    // `if_op_segment` refers to segments of IfRegions. `moved_op_list` refers to
    // the ops to be moved after new merged IfRegion. `regions_to_remove` refers to
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 17 07:31:01 UTC 2023
    - 25.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_communication.cc

        // we replace that block-argument's uses with the same (implicitly captured)
        // token 'token', used for If op, and erase the argument.
        // Note that 'RewriteControlFlowOpRegion' sets the token, used for the first
        // operation of region_idx'th region, to the dummy block-argument. As we
        // erase that argument, we also need to make sure that the token used for
        // the next operation is set to 'token'.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 40.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/replicate_to_island.cc

              op->getLoc(), DenseIntElementsAttr::get(
                                RankedTensorType::get({}, builder.getI64Type()),
                                {device_ordinal}));
          op->replaceAllUsesWith(const_op);
          op->erase();
          return WalkResult::advance();
        }
    
        if (!devices.has_value()) return WalkResult::advance();
    
        // Map aliased devices to explicit devices based on replica.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jul 24 21:01:40 UTC 2023
    - 16.9K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/breakup-islands.cc

          }
        }
        state.addOperands(operands);
        Operation* new_op = builder.create(state);
        item.replaceAllUsesWith(new_op);
        new_op->setAttrs(item.getAttrDictionary());
        item.erase();
      }
      AddStatefulControlFlowDependencies(graph_op);
    }
    
    // Populates an empty IslandOp and with a NoOp or Identity/IdentityN depending
    // on if there are any data results.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Aug 11 20:52:36 UTC 2023
    - 16.7K bytes
    - Viewed (0)
  6. 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)
  7. 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)
  8. tensorflow/compiler/mlir/lite/transforms/dense_to_sparse.cc

              auto s_const = builder.create<SparseConstOp>(
                  op->getLoc(), cst.getValue(), s_param, new_value);
              value.replaceAllUsesWith(s_const.getResult());
              cst.erase();
            } else if (type.getElementType().isF16()) {
              std::vector<Eigen::half> dense_data;
              dense_data.reserve(type.getNumElements());
              for (const auto& val : attr.getValues<APFloat>())
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 16.1K bytes
    - Viewed (0)
  9. tensorflow/c/eager/tape.h

      it->second--;
      if (it->second != 0) {
        return;
      }
      tensor_usage_.erase(it);
      auto tensor_op_it = tensor_tape_.find(tensor_id);
      if (tensor_op_it == tensor_tape_.end()) {
        return;
      }
      const int64_t op_id = tensor_op_it->second;
      if (op_id == -1) {
        // Do not delete watched tensors.
        return;
      }
      tensor_tape_.erase(tensor_op_it);
      auto op_it = op_tape_.find(op_id);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 02 12:40:29 UTC 2024
    - 47.2K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc

        }
        args.append(new_args.begin(), new_args.end());
      } else {
        args.append(yield_op->operand_begin(), yield_op->operand_end());
      }
      b.create<func::ReturnOp>(yield_op->getLoc(), args);
      yield_op->erase();
      SymbolTable(region.getParentOfType<ModuleOp>()).insert(outlined_func);
      outlined_func.setPrivate();
      return outlined_func;
    }
    
    // Replace region with call to outline function.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11.8K bytes
    - Viewed (0)
Back to top