- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for bitcast_op (0.17 sec)
-
tensorflow/c/kernels/BUILD
package( # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"], default_visibility = ["//visibility:public"], licenses = ["notice"], ) tf_kernel_library( name = "bitcast_op", prefix = "bitcast_op", deps = [ "//tensorflow/c:kernels", "//tensorflow/c:ops", "//tensorflow/c:tf_datatype", "//tensorflow/c:tf_status", "//tensorflow/c:tf_tensor",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 27 18:00:18 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/fold_broadcast_pass.cc
bcast_dims); } else { return rewriter.notifyMatchFailure(bcast_op, "Unsupported element type."); } Value new_const_op = rewriter.create<mhlo::ConstantOp>(bcast_op.getLoc(), result); rewriter.replaceOp(bcast_op, {new_const_op}); return success(); } }; using FoldBroadcastInDimBeforeMulOp =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.cc
Value add_op_result = op->getResult(0); // Broadcast bias value if unmatched with output shape. auto bcast_op = TryCast<BroadcastInDimOp>(bias_op.getDefiningOp(), /*name=*/"broadcast_in_dim_op"); if (failed(bcast_op)) { bcast_op = TryCast<DynamicBroadcastInDimOp>( bias_op.getDefiningOp(), /*name=*/"dynamic_broadcast_in_dim_op"); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 06:04:36 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize.cc
auto round_val = rewriter.create<TF::RoundOp>( sc_op.getLoc(), float_avg_pool_op.getOutput()); auto icast_op = rewriter.create<TF::CastOp>( sc_op.getLoc(), q_result_type.clone(qtype.getStorageType()), round_val); avg_pool_op.getResult().replaceAllUsesWith(icast_op.getResult()); return success(); } }; // Applies quantization on the model in TF dialect. class QuantizePass
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 23.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config.cc
TypeID::get<TF::AnyOp>(), TypeID::get<TF::AvgPoolOp>(), TypeID::get<TF::AvgPool3DGradOp>(), TypeID::get<TF::AvgPoolGradOp>(), TypeID::get<TF::BatchToSpaceNDOp>(), TypeID::get<TF::BitcastOp>(), TypeID::get<TF::BroadcastToOp>(), TypeID::get<TF::CollectivePermuteOp>(), TypeID::get<TF::ComplexOp>(), TypeID::get<TF::ConcatV2Op>(), TypeID::get<TF::ConjOp>(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 21.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
results.add<HashTableAndLookupTableFindToV2>(context); } //===----------------------------------------------------------------------===// // BitcastOp //===----------------------------------------------------------------------===// LogicalResult BitcastOp::verify() { BitcastOp op = *this; auto input_type = mlir::cast<ShapedType>(op.getInput().getType()); auto output_type = mlir::cast<ShapedType>(op.getOutput().getType());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
} if (type.isIntOrFloat()) { return std::max(type.getIntOrFloatBitWidth(), static_cast<unsigned>(CHAR_BIT)); } return -1; } LogicalResult BitcastOp::verify() { BitcastOp op = *this; auto input_type = op.getInput().getType().cast<ShapedType>(); auto output_type = op.getOutput().getType().cast<ShapedType>(); auto input_element_type = input_type.getElementType();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
for (auto [i, start_index] : llvm::enumerate(op.getStartIndices())) { // Start indices should be casted from tensor<i64> to tensor<1xi64>. auto cast = rewriter.create<TFL::BitcastOp>( op->getLoc(), single_element_type, start_index); int64_t upper_limit_idx = operand_shape[i] - slice_sizes[i]; auto upper_limit_attr =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0)