- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 206 for getBlob (0.13 sec)
-
tensorflow/compiler/mlir/lite/utils/tftext_utils.cc
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/tensorflow/transforms/functional_control_flow_to_regions.cc
assert(caller_region.empty() && "Expected empty region for newly created ops"); OpBuilder builder(caller_region); Block* entry = builder.createBlock(&caller_region); auto loc = op->getLoc(); if (use_region_args) { auto inputs = func.getFunctionType().getInputs(); entry->addArguments(inputs, SmallVector<Location>(inputs.size(), loc)); args = entry->getArguments(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0) -
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.
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/lite/transforms/while_loop_outline.cc
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()); } else {
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/quantization/stablehlo/passes/nchw_convolution_to_nhwc.cc
mlir::cast<TensorType>(input.getType()), kNchwToNhwcPermutation); auto input_transpose_op = rewriter.create<mlir::stablehlo::TransposeOp>( op.getLoc(), /*resultType0=*/new_input_tensor_type, /*operand=*/input, rewriter.getDenseI64ArrayAttr(kNchwToNhwcPermutation)); // Transpose the filter tensor: [o, i, 0, 1] => [0, 1, i, o] Value filter = op->getOperand(1);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc
} auto compile_op = builder->create<TF::_TPUCompileMlirOp>( cluster_func.getLoc(), /*compilation_status=*/compilation_status_type, /*program=*/ llvm::SmallVector<Type, 8>(num_cores_per_replica, program_type), compile_op_operands, txt_module, txt_metadata); return tensorflow::WrapOpInLaunch(builder, compile_op.getLoc(), compile_op, compilation_device); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 29.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
// Unformat op. auto unformat_op = builder.create<TF::TPUReshardVariablesOp>( while_op.getLoc(), llvm::ArrayRef<Type>{}, unformat_operands); WrapOpInLaunch(&builder, execute_launch.getLoc(), unformat_op, execute_launch.getDevice()); builder.create<tf_device::ReturnOp>(while_op.getLoc(), ArrayRef<Value>{}); return true; } void TPUVariableRuntimeReformattingPass::runOnOperation() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/xla_broadcast.cc
block_arg.getLoc(), DenseIntElementsAttr::get( RankedTensorType::get({1, num_replicas}, builder.getIntegerType(32)), group_assignment_val)); StringAttr reduce_op = builder.getStringAttr("Add"); StringAttr mode = builder.getStringAttr("CrossReplica"); return builder.create<XlaAllReduceOp>(block_arg.getLoc(), block_arg.getType(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 13.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/push_transpose_through_ewise.cc
auto *new_ewise_op = rewriter.create(op->getLoc(), op->getName().getIdentifier(), {tpose_arg1.getOperand(0), tpose_arg2.getOperand(0)}, new_out_type, op->getAttrs()); // Apply original tranpose to output of ewise op. auto out_tpose_op = rewriter.create<TFL::TransposeOp>( new_ewise_op->getLoc(), op->getResult(0).getType(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/fuse_convolution_pass.cc
} Value broadcast_multiplier = rewriter.create<mhlo::BroadcastInDimOp>( mul_op.getLoc(), filter.getType(), multiplier, broadcast_dims); Value new_filter = rewriter.create<mhlo::MulOp>( mul_op.getLoc(), filter.getType(), filter, broadcast_multiplier); Value new_conv = rewriter.create<mhlo::ConvolutionOp>( mul_op.getLoc(), conv_op.getType(), conv_op.getLhs(), new_filter,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 22:21:19 UTC 2024 - 8.3K bytes - Viewed (0)