- Sort Score
- Result 10 results
- Languages All
Results 11 - 14 of 14 for ConvertTensor (0.66 sec)
-
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_to_mhlo_int_test.cc
xla_literal->shape().element_type()) * xla_literal->element_count()); TF_ASSIGN_OR_RETURN(auto attrs, tensorflow::ConvertTensor(tensor, &builder)); builder.setInsertionPoint( &func_op.getFunctionBody().getBlocks().front().front()); // Use mhlo.Constant when it is consumed by the lowering passes since they
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 03 01:03:21 UTC 2024 - 35.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
} if (xla_arg.kind != XlaArgument::kConstant) { params.push_back(idx); continue; } TF_ASSIGN_OR_RETURN(auto value_attr, ConvertTensor(xla_arg.constant_value, &builder)); // TODO(hinsu): Use the actual location of the constant. auto constant = builder.create<mlir::TF::ConstOp>( mlir::UnknownLoc::get(module_op.getContext()), value_attr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
saved_model->variable_reader()->Lookup(checkpoint_key, &value), "Could not read checkpoint key from variables bundle: ", checkpoint_key); TF_ASSIGN_OR_RETURN(auto value_attr, ConvertTensor(value, &builder)); // A variable can have a partially known type, such as // tensor<?x27x?xf32>, even if the initializer is a specific static // shape.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)