- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 122 for Dtype (0.04 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/dump_graph_test.cc
} TEST(Dump, TextualIrWithOptions) { Graph graph(OpRegistry::Global()); Node* node; TF_ASSERT_OK(NodeBuilder("A", "Placeholder") .Attr("dtype", DT_FLOAT) .Finalize(&graph, &node)); string actual; StringWritableFile file(&actual); TF_ASSERT_OK(DumpTextualIRToFile(MlirDumpConfig().emit_location_information(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 03:47:51 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/aot/aot_only_var_handle_op.cc
.Attr("shared_name: string = ''") .Attr("debug_name: string = ''") .Attr("dtype: type") .Attr("shape: shape") .Output("resource: resource") .SetIsStateful() .SetShapeFn([](shape_inference::InferenceContext* c) { c->set_output(0, c->Scalar()); DataType t; TF_RETURN_IF_ERROR(c->GetAttr("dtype", &t)); PartialTensorShape p; TF_RETURN_IF_ERROR(c->GetAttr("shape", &p));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 28 09:57:04 UTC 2024 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
dtype == tensorflow::DT_UINT16 || dtype == tensorflow::DT_INT16 || dtype == tensorflow::DT_UINT8 || dtype == tensorflow::DT_INT8 || dtype == tensorflow::DT_HALF || dtype == tensorflow::DT_BFLOAT16 || dtype == tensorflow::DT_FLOAT || dtype == tensorflow::DT_DOUBLE || dtype == tensorflow::DT_COMPLEX64 || dtype == tensorflow::DT_COMPLEX128 || dtype == tensorflow::DT_BOOL) { return {};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
for (auto type : inst.getResultTypes()) if (!mlir::isa<mlir::TensorType, mlir::tf_executor::ControlType, mlir::tf_executor::TokenType>(type)) return errors::InvalidArgument( "Values must be of tensor type, TensorFlow control type, or " "TensorFlow token type. Found ", mlir::debugString(type));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
for (auto type : inst.getResultTypes()) if (!mlir::isa<mlir::TensorType, mlir::tf_executor::ControlType, mlir::tf_executor::TokenType>(type)) return errors::InvalidArgument( "Values must be of tensor type, TensorFlow control type, or " "TensorFlow token type. Found ", mlir::debugString(type));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util_test.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 19:51:50 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/c/eager/abstract_tensor_handle.cc
shape_string = "<error computing shape>"; } else { shape_string = shape.DebugString(); } return absl::StrCat("TensorHandle(shape=", shape_string, ", dtype=", DataType_Name(DataType()), ", type=", FullType().DebugString(), ")"); } Status AbstractTensorHandle::TensorHandleStatus() const { // Tensor handles in current runtime don't carry error info and this method
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 09:49:45 UTC 2024 - 1.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
tensorflow::CustomDevice* device, tensorflow::DataType dtype, void* data, TFE_CustomDeviceTensorHandleMethods methods) : tensorflow::CustomDeviceTensorHandle(context, device, dtype), data_(data), methods_(methods) {} ~CAPICustomDeviceTensorHandle() override { methods_.deallocator(data_); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 08:11:23 UTC 2024 - 44K bytes - Viewed (0) -
tensorflow/compiler/jit/get_compiler_ir.cc
const Tensor* value = variable.var()->tensor(); arg.type = value->dtype(); arg.shape = value->shape(); arg.initialized = true; } else { // Instead of embedding constant into HLO, // we handle tf.constant as parameter to reduce size. arg.kind = XlaCompiler::Argument::kParameter; arg.type = input->dtype(); arg.shape = input->shape(); } args.push_back(arg);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 19K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
void* data, size_t len, TF_Status* status) { auto dtype = static_cast<tensorflow::DataType>(data_type); DCHECK(tensorflow::DataTypeCanUseMemcpy(dtype)); tensorflow::Tensor tensor(dtype, tensorflow::TensorShape({})); std::memcpy(tensorflow::TensorCApi::Buffer(tensor)->data(), data, len); status->status = absl::OkStatus();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 29.4K bytes - Viewed (0)