- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 154 for Dtype (0.14 sec)
-
tensorflow/c/kernels.h
TF_Status* status); // Allocates a temporary Tensor of the specified type and shape. The // Tensor must not be used after kernel construction is // complete. // // num_dims must equal the size of array dims TF_CAPI_EXPORT extern TF_Tensor* TF_AllocateTemp( TF_OpKernelContext* context, TF_DataType dtype, const int64_t* dims, int num_dims, TF_AllocatorAttributes* alloc_attrs, TF_Status* status);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 09 22:46:22 UTC 2024 - 24.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_op_base.td
// Specify this trait if the op requires all outputs to have the same type and // the inputs either have the same type as result or a ref type corresponding to // the result type. def TF_OperandsSameAsResultsTypeOrRef : NativeOpTrait< "TF::OperandsSameAsResultsTypeOrRef">; // Op has the same operand and result element types (or type itself, if scalar) // after resolving reference types (i.e., after converting reference types to
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 30.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/BUILD
), ) tf_cc_binary( name = "bef_executor", testonly = True, visibility = [":friends"], deps = [ "@tf_runtime//:dtype", "@tf_runtime//:simple_tracing_sink", "@tf_runtime//tools:bef_executor_expensive_kernels", "@tf_runtime//tools:bef_executor_lib", "@tf_runtime//tools:bef_executor_lightweight_kernels",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 19:04:21 UTC 2024 - 23.6K bytes - Viewed (0) -
tensorflow/c/eager/gradients.cc
gtl::ArraySlice<AbstractTensorHandle*> output_gradients, absl::Span<AbstractTensorHandle*> result) const override; // Builds a tensor filled with ones with the same shape and dtype as `t`. Status BuildOnesLike(const TapeTensor& t, AbstractTensorHandle** result) const override; // Looks up the ID of a Gradient.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 09:49:45 UTC 2024 - 19.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental.h
// Summarizes the value of this tensor. The caller takes ownership of the // returned buffer. If `status` is not TF_OK, instead returns a null pointer. // // Does not include the shape and dtype of the tensor (which is generally // appended later), but should include any information specific to this custom // device which would be useful for debugging. //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 39.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/constant-fold.mlir
} // CHECK-LABEL: func @tfConst func.func @tfConst() -> (tensor<4xf32>, tensor<1x1x6x2xf32>) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 31 23:22:24 UTC 2024 - 36.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables.cc
ElementsAttr GetTensorValueAsElementsAttr( TF::VarHandleOp var_handle_op, const tensorflow::Tensor& resource_tensor, const tensorflow::DeviceMgr* mgr, OpBuilder builder) { if (resource_tensor.dtype() != tensorflow::DT_RESOURCE) { return GetTensorValueAsElementsAttr(resource_tensor, builder); } auto handle = resource_tensor.scalar<tensorflow::ResourceHandle>()();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 09:56:53 UTC 2024 - 19.4K bytes - Viewed (0) -
tensorflow/compiler/jit/compilability_check_util.cc
return false; } if (node.type_string() == "Const") { const AttrValue* attr = node.attrs().Find("dtype"); if (!op_filter_.allow_string_consts && attr != nullptr && attr->type() == DT_STRING) { *uncompilable_reason = "Const op with type DT_STRING is not supported by XLA."; return false; } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 30.3K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc
for (int i = 0; i < 2; ++i) { tensor_proto.add_int_val(1); } NodeDef const_def; TF_CHECK_OK(NodeDefBuilder("const", "Const") .Attr("dtype", DT_INT32) .Attr("value", tensor_proto) .Finalize(&const_def)); Status s; Node *const_node = g->AddNode(const_def, &s); TF_CHECK_OK(s); NodeDef fn_def;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 41K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_to_mhlo_int_test.cc
llvm::dyn_cast<TF::ConstOp>(fold_results[0].getDefiningOp()).getValue(), &tensor)); xla::Shape xla_shape; TF_RETURN_IF_ERROR(tensorflow::TensorShapeToXLAShape( tensor.dtype(), tensor.shape(), &xla_shape)); xla::PjRtClient::HostBufferSemantics host_buffer_semantics = xla::PjRtClient::HostBufferSemantics::kImmutableUntilTransferCompletes; TF_ASSIGN_OR_RETURN( auto buffer,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 03 01:03:21 UTC 2024 - 35.8K bytes - Viewed (0)