- Sort Score
- Result 10 results
- Languages All
Results 11 - 15 of 15 for int64Ptr (0.12 sec)
-
tensorflow/compiler/mlir/lite/quantization/lite/quantize_model_test.cc
const std::vector<float>& bias_scales = bias_tensor->quantization->scale; const std::vector<float>& weights_scales = weights_tensor->quantization->scale; const std::vector<int64_t>& weights_zero_points = weights_tensor->quantization->zero_point; const int out_channel_size = weights_tensor->shape[0]; ASSERT_THAT(bias_scales, SizeIs(out_channel_size));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 23:15:24 UTC 2024 - 73.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
"ranked tensors."; ArrayRef<int64_t> in_shape = ranked_type.getShape(); if (in_shape.empty() || in_shape[0] < 0) { return context_op->emitOpError() << "A map_outside_compilation op's input and output shapes must " "have rank at least one and the first dimension must be known."; } int64_t split_size = in_shape[0] / num_cores_per_replica;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_import.cc
{min_maxs[0], min_maxs[1]}); mlir::ElementsAttr axis_stats; mlir::IntegerAttr axis; if (mins.size() > 1) { llvm::SmallVector<int64_t, 4> axis_stats_shape{ static_cast<int64_t>(mins.size()), 2}; axis_stats = mlir::DenseFPElementsAttr::get( tensorflow::GetTypeFromTFTensorShape(axis_stats_shape, b.getF32Type()), min_maxs);
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/jit/deadness_analysis_test.cc
EXPECT_EQ(predicate_map[ControlOutputFor(id_true)], "*cond_ref:0"); } void CreateSwitchN(const Scope& scope, Input data, Input output_index, int64_t num_outs, OutputList* outputs) { if (!scope.ok()) return; auto _data = ops::AsNodeOut(scope, data); if (!scope.ok()) return; auto _output_index = ops::AsNodeOut(scope, output_index);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 51.6K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass_test.cc
Output delta = ops::Placeholder(scope.WithOpName("delta"), DT_FLOAT); ops::AssignAddVariableOp increment_op( scope.WithOpName("IncrementIteration"), iter, ops::Const(scope.WithOpName("one"), static_cast<int64_t>(1))); ops::AssignAddVariableOp weights_0_update_op( scope.WithOpName("weights_0_update"), weights_0, delta); ops::AssignAddVariableOp weights_1_update_op(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 14 10:11:10 UTC 2024 - 79.6K bytes - Viewed (0)