- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 176 for getBlob (0.32 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/rewrite_tpu_embedding_ops.cc
SendTPUEmbeddingGradientsOp send_op; if (failed(GetOp(region, &send_op))) return failure(); // No TPU embedding ops. if (!recv_op && !send_op) return success(); Location loc = recv_op ? recv_op.getLoc() : send_op.getLoc(); StringRef config = recv_op ? recv_op.getConfig() : send_op.getConfig(); // Create XlaRecvTPUEmbeddingDeduplicationData op. OpBuilder builder(region); auto output_ty =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 22:55:42 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_outlining.cc
void runOnOperation() override; }; void ReplaceClusterReturnWithReturn(tf_device::ReturnOp cluster_return_op, OpBuilder* builder) { builder->create<func::ReturnOp>(cluster_return_op.getLoc(), cluster_return_op.getOperands()); cluster_return_op.erase(); } // Builds a function that outlines region attached to cluster_op or launch_op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_globals_to_ml_program.cc
} else { initial_value = mlir::Attribute(); } opToName[globalTensor] = name; auto variableOp = globalBuilder.create<ml_program::GlobalOp>( globalTensor.getLoc(), name, globalTensor.getType(), globalTensor.getIsMutable(), initial_value, /*visibility=*/globalBuilder.getStringAttr("private")); variableOp.setPrivate(); } SymbolTable syms(module);
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/tf2xla/transforms/legalize_tf_communication.cc
SmallVector<Value, 4> tuple_operands; for (auto idx : llvm::seq<int32_t>(0, end)) tuple_operands.push_back( builder.create<GetTupleElementOp>(value.getLoc(), value, idx) .getResult()); return CreateTuple(builder, value.getLoc(), tuple_operands); } // Replaces uses of `values` with `replacements`. If `flatten_tuple` is false,
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/mlir/lite/quantization/ir/ConvertSimQuant.cc
// TODO: Map to a qbarrier with an attribute like [Forced] to signal that // this is a forced/hard-coded constraint. auto qbarrier = rewriter.create<QuantizeCastOp>(op.getLoc(), quantizedType, op.getInputs()); rewriter.replaceOpWithNewOp<DequantizeCastOp>(op, converter.input_type,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 02:10:16 UTC 2024 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_partitioned_op_conversion.cc
auto pi = builder.create<TF::TPUPartitionedInputV2Op>( op.getLoc(), op.getType(), op.getOperands(), builder.getI64ArrayAttr(partition_dims), builder.getBoolAttr(false), op.get_XlaShardingAttr()); op->replaceAllUsesWith(pi); } else { auto po = builder.create<TF::TPUPartitionedOutputV2Op>( op.getLoc(), op.getResultTypes(), op.getOperand(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
FilterOutBlockArgControlDep(op->getOperands().drop_front(), new_operands); auto source_op = builder.create<tf_executor::NextIterationSourceOp>( op->getLoc(), op->getOperand(0).getType()); builder.create<tf_executor::NextIterationSinkOp>( op->getLoc(), source_op.getToken(), /*input=*/op->getOperand(0), /*controlInputs=*/new_operands); op->replaceAllUsesWith(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/perception_ops_utils.cc
return failure(); } auto op = builder.create<CustomOp>( func_.getLoc(), func_.getFunctionType().getResults(), func_.getArguments(), kMaxUnpooling, CustomOption(&builder, custom_option_buffer)); builder.create<func::ReturnOp>(func_.getLoc(), op.getResults()); return success(); } LogicalResult ConvertMaxUnpoolingFunc::VerifySignature() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_xla_attribute_utils_test.cc
OpBuilder builder(&module->getBodyRegion()); context.loadDialect<TF::TensorFlowDialect>(); Value value = CreateConstValue<int8_t>(builder, module->getLoc(), unpacked_shape, unpacked_values); Value packed_value = PackOperand(builder, module->getLoc(), value, pack_dim); DenseIntElementsAttr packed_value_attr; ASSERT_TRUE(matchPattern(packed_value, m_Constant(&packed_value_attr)));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h
maxs.push_back(rmax); } quant_type = quantfork::fakeQuantAttrsToType( op.getLoc(), num_bits, *op.getAxis(), mins, maxs, narrow_range, expressed, is_signed); if (legacy_float_scale) { quant_type = DownCastScale(quant_type, mins, maxs, op->getLoc()); } } else if (auto stats = op.getLayerStats().dyn_cast<DenseFPElementsAttr>()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 41.7K bytes - Viewed (0)