- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 263 for created (0.08 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_asset_sinking_pass.cc
// Create SessionInitializerOp; should reference main function. StringAttr func_name = main_func.getSymNameAttr(); llvm::SmallVector<mlir::Attribute, 2> func_names = { mlir::SymbolRefAttr::get(builder.getContext(), func_name)}; builder.create<tf_saved_model::SessionInitializerOp>( module->getLoc(), builder.getArrayAttr(func_names)); // Create AssetOp; this holds the checkpoint_path.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tf_xla_op_to_tf_op.cc
indices[i] = dimension_numbers.start_index_map()[i]; } // Fill elements from start_indices with start_index_map Value scattered_start_indices = builder.create<TF::TensorScatterUpdateOp>( loc, empty_start_indices, /*indices=*/ builder.create<TF::ReshapeOp>( loc, RankedTensorType::get({index_map_size, 1}, builder.getI64Type()), Create1DConstValue<int64_t>(builder, loc, indices),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 13.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
Value new_reshape_shape_value = rewriter.create<arith::ConstantOp>( inner_reshape_op->getLoc(), GetI32ElementsAttr(new_reshape_shape_i32, &rewriter)); auto new_inner_reshape_op = rewriter.create<TFL::ReshapeOp>( inner_reshape_op->getLoc(), inner_reshape_input, new_reshape_shape_value); // Create a new reshape_op to replace the old inner reshape_op.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/unroll_batch_matmul.cc
Type resultType = RankedTensorType::get(shape, element_type); auto constant_attr = DenseElementsAttr::get(shape_spec_type, shape); auto shape_tensor = rewriter.create<TF::ConstOp>(loc, shape_spec_type, constant_attr); return rewriter.create<TF::ReshapeOp>(loc, resultType, /*tensor=*/value, /*shape=*/shape_tensor); } template <typename BatchMatMulOpType>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc
// Creates a _Retval node for the src node of edge, and add it to results_, // if none exists yet. If a new _Retval node is created, also adds the edge // within the subgraph from the src to the _Retval node. Status RecordResult( const Edge* edge, const absl::flat_hash_map<const Node*, Node*>& node_images);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 51K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/utils/stablehlo_type_utils_test.cc
const OwningOpRef<mlir::stablehlo::ConstantOp> constant_op = builder_.create<mlir::stablehlo::ConstantOp>( builder_.getUnknownLoc(), builder_.getI32IntegerAttr(0)); EXPECT_TRUE(IsStablehloOp(*constant_op)); } TEST_F(StablehloTypeUtilsTest, IsStablehloOpFailsWithArithOp) { const OwningOpRef<mlir::arith::ConstantOp> constant_op = builder_.create<mlir::arith::ConstantOp>(builder_.getUnknownLoc(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 25 00:32:20 UTC 2024 - 2.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/utils/dialect_detection_utils_test.cc
context.loadDialect<TensorFlowDialect>(); OpBuilder opBuilder(&context); OperationState state(UnknownLoc::get(opBuilder.getContext()), /*OperationName=*/"tf.Const"); mlir::Operation* op = Operation::create(state); bool result = IsInBridgeAcceptableDialects(op); EXPECT_TRUE(result); op->destroy(); } TEST_F(SharedUtilsTest, IsInFunctionalDialectFails) { MLIRContext context;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 08 20:33:37 UTC 2023 - 2.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/tftext_utils.cc
func->setAttr(kTFImplements, attr); OpBuilder builder(func.getBody()); std::string empty_option_buffer; auto op = builder.create<CustomOp>( func.getLoc(), func.getFunctionType().getResults(), func.getArguments(), api, CustomOption(&builder, empty_option_buffer)); builder.create<func::ReturnOp>(func.getLoc(), op.getResults()); return success(); } LogicalResult VerifyNgrams(func::FuncOp func) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/io.cc
namespace stablehlo::quantization::io { absl::StatusOr<std::string> GetLocalTmpFileName(tsl::Env* const env) { std::string tmp_fname{}; if (!env->LocalTempFilename(&tmp_fname)) { return absl::InternalError("Failed to create tmp file name."); } return tmp_fname; } absl::StatusOr<std::string> GetLocalTmpFileName() { return GetLocalTmpFileName(tsl::Env::Default()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 03:28:15 UTC 2024 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/io_test.cc
// A test-only derived class of `tsl::Env` which is broken. Used to cause // failure for the `CreateTmpDir` function. Each of the overridden member // functions implements a dummy functionality just to be able to create an // instance of this class. class TestEnvBrokenFileSystem : public tsl::Env { public: TestEnvBrokenFileSystem() = default; bool MatchPath(const tsl::string& path, const tsl::string& pattern) override {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 03:28:15 UTC 2024 - 6.7K bytes - Viewed (0)