- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 442 for pushBack (0.15 sec)
-
tensorflow/compiler/mlir/lite/transforms/dense_to_sparse.cc
*b_map = {}; *b_size = {}; int block_rank = 0; for (int i = 0; i < dims_count; i++) { if (block_size[i] != 1) { traversal_order->push_back(block_rank + dims_count); format->push_back(kTfLiteDimDense); block_rank++; b_map->push_back(i); b_size->push_back(block_size[i]); } } } inline float GetSparsity(const int num_zeros, const int num_elements) {
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/jit/compilation_passes_test_main.cc
args.reserve(real_argc + 1); for (int i = 0; i < real_argc; i++) { args.push_back(real_argv[i]); } struct FreeDeleter { void operator()(char* ptr) { free(ptr); } }; std::unique_ptr<char, FreeDeleter> enable_global_jit_arg( strdup("--tf_xla_cpu_global_jit=true")); args.push_back(enable_global_jit_arg.get()); std::unique_ptr<char, FreeDeleter> reduce_min_cluster_size_arg(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 14 00:31:28 UTC 2019 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/decompose_hybrid_quantization.cc
auto newTy = QuantizedType::castToExpressedType(operand.getType()); newOperands.push_back( rewriter.create<TFL::DequantizeOp>(loc, newTy, operand)); continue; } newOperands.push_back(operand); } SmallVector<Type> newResultTys; for (auto result : op->getResults()) { Type resultTy = result.getType();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.8K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/test_utils.cc
result.reserve(set.size()); for (DataType dt : set) { result.push_back(dt); } return result; } std::vector<std::vector<int64_t>> InterestingShapes() { std::vector<std::vector<int64_t>> interesting_shapes; interesting_shapes.push_back({}); // Scalar interesting_shapes.push_back({10}); // 1D Vector interesting_shapes.push_back({3, 3}); // 2D Matrix
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Sep 08 20:13:32 UTC 2023 - 5.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
padded_begin.push_back(0); padded_end.push_back(0); padded_stride.push_back(1); } // Account for ellipsis mask. ++index; // After the ellipsis. for (; index < begin_shape[0];) { padded_begin.push_back(begin_dense_elem_attr.getValues<int32_t>()[index]); padded_end.push_back(end_dense_elem_attr.getValues<int32_t>()[index]);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/dot_general.cc
for (const int64_t dim : batch_dimensions) { batch_dimensions_.axes.push_back(dim); batch_dimensions_.sizes.push_back(type.getDimSize(dim)); } for (const int64_t dim : contracting_dimensions) { contracting_dimensions_.axes.push_back(dim); contracting_dimensions_.sizes.push_back(type.getDimSize(dim)); } for (int64_t dim = 0; dim < rank; ++dim) {
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/mlir/quantization/stablehlo/passes/quantization_patterns.h
if (mlir::isa<NoneType>(operand_type)) { inputs.push_back(operand); continue; } auto ele_type = mlir::cast<TensorType>(operand.getType()).getElementType(); if (auto dq_op = dyn_cast_or_null<DequantizeOpT>(operand.getDefiningOp())) { inputs.push_back(dq_op.getOperand()); } else if (!ele_type.isF32()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc
}); std::vector<std::string> lines; lines.push_back("-------- Quantization Summary --------"); lines.push_back("Number of quantized layers in the model"); lines.push_back("--------------------------------"); lines.push_back( absl::StrFormat("%s Count/Total", pad_string("Name", name_col_width))); lines.push_back("================================");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 54.5K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util_gpu_test.cc
}; TEST_F(PjRtExecutionUtilGpuTest, PreparePjRtExecutableArguments) { std::vector<const Tensor*> inputs; inputs.push_back(CreateDeviceTensor<int32_t>(TensorShape({1, 3}), {0, 0, 0})); inputs.push_back(CreateDeviceTensor<int32_t>(TensorShape({1, 3}), {1, 2, 3})); inputs.push_back(CreateDeviceTensor<int32_t>(TensorShape({1, 3}), {4, 5, 6})); int num_missing_prefix_ctx_inputs = 2; std::vector<int> input_mapping{3, 4};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 10K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/insert_calibration_statistics_saver.cc
if (aggregator_ops_to_ignore.count(op.getId().str())) continue; ids.push_back(op.getId()); calibration_methods.push_back(op.getCalibrationMethod()); statistics_outputs.push_back(op.getMin()); statistics_outputs.push_back(op.getMax()); statistics_outputs.push_back(op.getHistogram()); } } // Inserts a `CalibrationStatisticsSaverOp` to the end of the region.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 7.2K bytes - Viewed (0)