Search Options

Results per page
Sort
Preferred Languages
Advance

Results 121 - 130 of 270 for getFlow (0.16 sec)

  1. tensorflow/compiler/mlir/lite/transforms/partitioned_topological_sort.cc

          ++num_unscheduled_ops;
        }
      }
      emitRemark(func.getLoc(), func.getName())
          << ": " << num_part_ops << " ops delegated out of " << num_all_ops
          << " ops with " << num_partitions_after
          << " partitions (originally: " << num_partitions_before << ")";
      if (!scheduled_everything) {
        emitError(func.getLoc(), func.getName())
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Dec 19 15:05:28 UTC 2023
    - 7.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/readonly_references_to_resources.cc

        if (variable_name.empty()) {
          return signalPassFailure();
        }
        VarHandleOp var_handle_op = builder.create<VarHandleOp>(
            variable_v2_op.getLoc(),
            ArrayRef<Type>{RankedTensorType::get(
                {}, TF::ResourceType::get(ArrayRef<TensorType>{tensor_type},
                                          builder.getContext()))},
            ArrayRef<Value>{},
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/tensorflow/utils/fake_quant_utils.h

        // and its users.
        auto quantize = rewriter.create<quantfork::QuantizeCastOp>(
            tf_op.getLoc(), qtype.getValue(), input);
        auto dequantize = rewriter.create<quantfork::DequantizeCastOp>(
            tf_op.getLoc(), res_type, quantize.getResult());
        tf_op.getOutputs().replaceAllUsesWith(dequantize);
    
        return success();
      }
    
      bool use_fake_quant_num_bits_;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 6.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_stablehlo_composite_to_tfl_custom.cc

      if (IsKVCacheCompositeOp(composite)) {
        return builder.create<TFL::CustomOp>(
            composite->getLoc(), composite->getResultTypes(),
            composite->getOperands().slice(2, 3), composite.getName(),
            CustomOption(&builder, custom_option_buffer));
      }
      return builder.create<TFL::CustomOp>(
          composite->getLoc(), composite->getResultTypes(),
          composite->getOperands(), composite.getName(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 6.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/utils/fake_quant_utils.h

        // and its users.
        Value value = tf_op.getOutputs();
        auto quantize = rewriter.create<TFL::QuantizeOp>(
            tf_op.getLoc(), qtype.getValue(), value, qtype);
        auto dequantize = rewriter.create<TFL::DequantizeOp>(
            tf_op.getLoc(), res_type, quantize.getOutput());
        value.replaceAllUsesWith(dequantize);
        quantize.getOperation()->replaceUsesOfWith(dequantize, value);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 6.6K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/transforms/insert_call_once_op.cc

          if (!dict_attr) continue;
    
          OpBuilder builder(func.getContext());
          builder.setInsertionPointToStart(&func.getBlocks().front());
          builder.create<TFL::CallOnceOp>(func.getLoc(), init_func_op.getName());
        }
      }
    }
    
    }  // namespace
    
    // Inserts a TFL::CallOnce op when tf_saved_model's session initializer is
    // given.
    std::unique_ptr<OperationPass<ModuleOp>>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 01 05:03:09 UTC 2023
    - 2.5K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_async.cc

                  operand.getDefiningOp<corert::ConstDenseTensorOp>()) {
            new_values.push_back(
                rewriter.create<fallback_async::ConstDenseTensorOp>(
                    op.getLoc(), rewriter.getType<fallback::TFTensorType>(),
                    corert_const_dense_tensor_op.getValue()));
            should_rewrite = true;
            continue;
          }
          if (auto corert_const_string_tensor_op =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 08 01:19:25 UTC 2023
    - 15.6K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/transforms/executor_island_coarsening.cc

        }
      }
      auto island = OpBuilder(fetch).create<IslandOp>(
          fetch.getLoc(), data_types,
          /*control=*/ControlType::get(fetch.getContext()),
          /*controlInputs=*/control_fetches);
      island.getBody().push_back(new Block);
      OpBuilder::atBlockEnd(&island.GetBody())
          .create<YieldOp>(fetch.getLoc(), data_fetches);
      const int fetch_control_idx = data_fetches.size();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc

            // and once existing cases have been fixed.
            mlir::InFlightDiagnostic error =
                strict_clusters ? mlir::emitError(op.getLoc(), "")
                                : mlir::emitWarning(op.getLoc(), "");
            error << "Op has cyclic dependency with a compilation cluster:\n";
            error << "The cluster depends on\n";
            error << op.getName() << "\n"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 22:03:30 UTC 2024
    - 39.3K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_merge_variables_with_execute.cc

      auto merged_execute_launch = builder->create<tf_device::LaunchOp>(
          merged_execute.getLoc(), execute_launch.getDeviceAttr(),
          merged_execute.getResultTypes());
      merged_execute_launch.getBody().push_back(new Block);
    
      builder->setInsertionPointToEnd(&merged_execute_launch.GetBody());
      builder->create<tf_device::ReturnOp>(merged_execute.getLoc(),
                                           merged_execute.getResults());
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 29 17:52:11 UTC 2024
    - 27K bytes
    - Viewed (0)
Back to top