- Sort Score
- Result 10 results
- Languages All
Results 91 - 100 of 182 for created (0.17 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc
auto reshape_sizes = DenseIntElementsAttr::get(reshape_type, new_shape); auto reshape_value = builder->create<TF::ConstOp>(input.getLoc(), reshape_sizes); return builder->create<TF::ReshapeOp>(input.getLoc(), reshape_result_type, input, reshape_value); } // Creates transpose op for shape to depth transform. TF::TransposeOp GetTransposeOpForConv2DFilter(OpBuilder* builder, Value input) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 29.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/lift_quantizable_spots_as_functions.cc
if (failed(ApplyQuantizationSpec(spec, module_op))) { signalPassFailure(); return; } } } } // namespace // Creates `LiftQuantizableSpotsAsFunctionsPass` with user-defined // `QuantizationSpecs`. std::unique_ptr<OperationPass<ModuleOp>> CreateLiftQuantizableSpotsAsFunctionsPass( const QuantizationSpecs& quantization_specs) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 9.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc
} else { if (IsCompatibleTypeWithTFLCastOp(value.getType()) && IsCompatibleTypeWithTFLCastOp(type)) { auto cast = b.create<CastOp>(yield_op->getLoc(), type, value); args.push_back(cast); } else { auto cast = b.create<TF::CastOp>(yield_op->getLoc(), type, value); args.push_back(cast); } } } args.append(new_args.begin(), new_args.end());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_control_to_data_outputs.cc
} // Creates a wrapper island enclosing the `sub_op` dependent on // `control_inputs`. IslandOp CreateIsland(Operation* sub_op, ValueRange control_inputs, OpBuilder builder) { assert(sub_op); auto control_type = ControlType::get(builder.getContext()); auto island = builder.create<IslandOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/dense_to_sparse.cc
auto s_qconst = builder.create<SparseQConstOp>(op->getLoc(), cst.getQtypeAttr(), cst.getValue(), s_param, new_value); value.replaceAllUsesWith(s_qconst.getResult()); cst.erase(); } if (result.needs_densify) { auto value = op->getOperand(operand); auto densify = builder.create<DensifyOp>(op->getLoc(), value.getType(), value);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/dot_general.cc
RankedTensorType::get(static_cast<int32_t>(1), builder.getI32Type()); auto out_segids_cst = builder.create<TFL::ConstOp>( builder.getI32TensorAttr(flattened_out_segids)); auto contracting_segids_cst = builder.create<TFL::ConstOp>( builder.getI32TensorAttr(flattened_contracting_segids)); auto num_segids_tensor = builder.create<TFL::ConstOp>(DenseIntElementsAttr::get(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.2K bytes - Viewed (0) -
tensorflow/compiler/jit/flags.cc
"Disable deadness related safety checks when clustering (this is " "unsound)."), Flag("tf_xla_disable_resource_variable_safety_checks_for_debugging", &mark_for_compilation_flags ->tf_xla_disable_resource_variable_safety_checks_for_debugging, "Disable resource variables related safety checks when clustering " "(this is unsound)."),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 18:52:57 UTC 2024 - 24.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_import.cc
if (!shaped_type) { return errors::Internal("Constant doesn't have a shape"); } auto op = builder.create<tfl::ExternalConstOp>( loc, shaped_type, builder.getI32IntegerAttr(buffer_index)); return op.getOperation(); } // TODO(b/172664358): Creates a new op instead of reusing constant op. // Creates a constant op with "tfl.is_variable" attribute to represent stateful
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_ops_by_policy.cc
return a->isBeforeInBlock(b); }); // Create tf_device::ClusterOp before the last operation in the block that // is a part of a match set. auto back = cluster.operations.back(); auto loc = back->getLoc(); OpBuilder builder(back); auto cluster_op = builder.create<tf_device::ClusterOp>(loc, return_types, policy); // Create block in cluster_op's region and move 'cluster.operations' into
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 27.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
auto slice_size_attr = DenseIntElementsAttr::get(idx_type, slice_size_i32); auto slice_size = rewriter.create<arith::ConstantOp>(op.getLoc(), slice_size_attr); ArrayRef<int64_t> strides = op.getStrides(); // If stride of every dimension is 1, create tfl.slice and return early. // Otherwise, create tfl.strided_slice instead. if (llvm::all_of(strides, [](int64_t stride) { return stride == 1; })) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0)