- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 3,700 for Reserve (0.2 sec)
-
tensorflow/compiler/mlir/lite/transforms/legalize_tensorlist.cc
std::optional<ConstBytesAttr> CustomOptions(MLIRContext* context, mlir::Operation* op) { if (auto reserve = llvm::dyn_cast_or_null<mlir::TF::TensorListReserveOp>(op)) { tflite::TensorType tflite_type = tflite::ConvertTypeToTensorType(reserve.getElementDtype()); return CreateListReserveOptions(context, tflite_type); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 23:04:40 UTC 2024 - 10.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/const_tensor_utils.cc
llvm::SmallVector<mlir::APInt> ret; size_t read_size = sizeof(T); int bytes_len = bytes.size(); assert(bytes_len % read_size == 0); int elem_count = bytes_len / read_size; ret.reserve(elem_count); const char* data_ptr = reinterpret_cast<const char*>(bytes.data()); for (int i = 0; i < elem_count; i++) { T val = llvm::support::endian::readNext<T, llvm::endianness::native,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 23:04:40 UTC 2024 - 16.6K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util_test.cc
device_->parsed_name().id)); std::vector<xla::PjRtBuffer*> executable_args; executable_args.reserve(result->input_mapping.size()); absl::flat_hash_set<int> non_donatable_input_indices; TF_EXPECT_OK(PreparePjRtExecutableArguments( /*num_missing_prefix_ctx_inputs=*/0, result->input_mapping, inputs,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 28.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc
const mlir::Value& original_source, mlir::OpBuilder* builder, llvm::SmallVectorImpl<mlir::Value>* tiled_inputs) { llvm::SmallVector<mlir::TF::SplitOp, 4> split_ops_for_tiled_input; split_ops_for_tiled_input.reserve( input_sharding.tile_assignment_devices_size()); // Creates a tree of split nodes for sharding tiled inputs. Splits nodes // are created such that input data is sharded in row major order.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 34K bytes - Viewed (0) -
pkg/scheduler/testing/framework/fake_plugins.go
}, nil } } // FakeReservePlugin is a test reserve plugin. type FakeReservePlugin struct { Status *framework.Status } // Name returns name of the plugin. func (pl *FakeReservePlugin) Name() string { return "FakeReserve" } // Reserve invoked at the Reserve extension point. func (pl *FakeReservePlugin) Reserve(_ context.Context, _ *framework.CycleState, _ *v1.Pod, _ string) *framework.Status {
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Mon Feb 26 19:07:19 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/odml_converter/folders.cc
} return FoldAdaptor(operation); } // Gets a list of ElementsAttr behind each constant operand. llvm::SmallVector<ElementsAttr> OperandData() { llvm::SmallVector<ElementsAttr> res; res.reserve(operation_->getNumOperands()); for (auto opr : operation_->getOperands()) { auto op = llvm::dyn_cast<stablehlo::ConstantOp>(opr.getDefiningOp()); res.push_back(op.getValue()); } return res; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 06:11:55 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/convert_tensor.cc
ShapedType type) { // Extract to a vector of StringRefs for converting. auto arr = input_tensor.flat<tstring>(); std::vector<mlir::StringRef> string_refs; string_refs.reserve(arr.size()); for (int i = 0; i < arr.size(); i++) { const auto& val = arr(i); string_refs.push_back({val.data(), val.size()}); } return ElementsAttr(DenseStringElementsAttr::get(type, string_refs));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 20.5K bytes - Viewed (0) -
tensorflow/c/eager/unified_api_testutil.cc
absl::flat_hash_set<int> null_indices; { AbstractContextPtr func_ctx(BuildFunction(fn_name)); std::vector<AbstractTensorHandle*> func_inputs; func_inputs.reserve(inputs.size()); TF_RETURN_IF_ERROR( CreateParamsForInputs(func_ctx.get(), inputs, &func_inputs)); std::vector<AbstractTensorHandle*> model_outputs; model_outputs.resize(outputs.size());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 27 13:57:45 UTC 2024 - 5.7K bytes - Viewed (0) -
pkg/scheduler/metrics/metrics.go
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Sat Apr 27 08:22:53 UTC 2024 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/ir/ConvertSimQuant.cc
quant::QuantizedType convertFakeQuantAttrsToType(ConstFakeQuantPerAxis fqOp, Type expressedType) const { SmallVector<double, 4> min, max; min.reserve(fqOp.getMin().size()); max.reserve(fqOp.getMax().size()); for (auto m : fqOp.getMin()) min.push_back(cast<FloatAttr>(m).getValueAsDouble()); for (auto m : fqOp.getMax())
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 02:10:16 UTC 2024 - 6K bytes - Viewed (0)