- Sort Score
- Result 10 results
- Languages All
Results 121 - 130 of 270 for getFlow (0.16 sec)
-
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 39.3K bytes - Viewed (0) -
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)