- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 235 for created (0.13 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/instrumentations/save_report_test.cc
report_file_path)); const LogicalResult run_result = pm.run(*module_op); ASSERT_TRUE(succeeded(run_result)); // The report file is not created because `QuantizeCompositeFunctionsPass` was // not run. EXPECT_THAT(ReadFileToString(report_file_path), StatusIs(absl::StatusCode::kNotFound)); } TEST_F(SaveQuantizationReportInstrumentationTest,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 02:59:01 UTC 2024 - 9.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/quantize_passes.cc
mlir::quant::stablehlo::FillPresetQuantizationOptions( quantization_options); } // TODO: b/276999414 - Add activation and bias quantization component as // respective quantization passes are created. QuantizationComponentSpec weight_component; for (const auto& component : quantization_options_.quantization_method() .custom_quantization_method()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 08:32:43 UTC 2024 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/clustering_bridge_passes.cc
// Guarantee all functions have one use, which enables more exact shape // inference. pm.addPass(mlir::TF::CreateGuaranteeAllFuncsOneUsePass()); // Run shape inference so that tf_executor/tf_device ops created later will // likely to inherit more concrete types. pm.addPass(mlir::TF::CreateTFShapeInferencePass()); pm.addNestedPass<FuncOp>(mlir::TFTPU::CreateTPUPartitionedOpConversionPass()); pm.addNestedPass<FuncOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 16:09:14 UTC 2024 - 11.2K bytes - Viewed (0) -
tensorflow/c/c_api.cc
"and will trigger an error in the future. Either don't modify " "nodes after running them or create a new session."); } } } namespace { // Helper method that creates a shape handle for a shape described by dims. tensorflow::shape_inference::ShapeHandle ShapeHandleFromDims( tensorflow::shape_inference::InferenceContext* ic, int num_dims,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_communication.cc
return tokens[0]; } else { return builder.create<AfterAllOp>(loc, original_token.getType(), tokens) .getResult(); } } // Replaces `tf._XlaHostComputeMlir` with individual `mhlo.send` and `mhlo.recv` // ops per operand and result. Unique Channel IDs are assigned per transfer. // Sink tokens are created across all `mhlo.send` ops first and then by // all `mhlo.recv` ops.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 40.5K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util.cc
if (pjrt_tensor_buffer != nullptr) { args->push_back(pjrt_tensor_buffer->pjrt_buffer()); } else { // Creates a PjRtBuffer from DeviceMemoryBase. The newly created // PjRtBuffer needs to be persisted till XLA execution is completed. auto dmem = se::DeviceMemoryBase( const_cast<char*>(tensor->tensor_data().data()),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 40.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/extract_tpu_copy_with_dynamic_shape_op.cc
auto new_launch_op = builder.create<tf_device::LaunchOp>( old_launch_op->getLoc(), old_launch_op->getDeviceAttr(), /*result_types=*/new_launch_op_results_types); new_launch_op.getBody().takeBody(old_launch_op->getBody()); new_launch_op.GetBody().getTerminator()->setOperands(new_launch_op_results); return new_launch_op; } // Create the new device launch op which wraps the copy op.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/convert_func_to_bfloat16.cc
new_results))) return failure(); // Update the regions. The dialect conversion framework wants new regions to // be created and updated, rather than updating the old op. Thus we use an // OperationState so we can add regions to the new op. OperationState state(op->getLoc(), op->getName().getStringRef(), operands,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_outline_tpu_island.cc
OpBuilder replacer(yield_op); island_op.getBody().push_back(new Block); replacer.create<mlir::func::ReturnOp>(yield_op.getLoc(), yield_op.getOperands()); yield_op.erase(); } // Remap the captured operands in the (former) island block with newly // created entry block arguments in the function body. {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_head_tail_outside_compilation.cc
} // Wraps block in a Launch. External uses of ops in the block will be return // values of the Launch and remapped to the Launch results. If `before` is set // to true, the Launch is created before `op`. Otherwise the Launch is created // after `op`. mlir::tf_device::LaunchOp CreateLaunchForBlock(OpBuilder* builder, Operation* op, bool before,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20.6K bytes - Viewed (0)