- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 155 for Dtype (0.07 sec)
-
tensorflow/compiler/mlir/tfrt/tests/hoist_invariant_ops.mlir
// CHECK-NEXT: "tf.AssignVariable"([[v]]#[[handle_idx]], [[v]]#[[const_idx]]) {device = "/CPU:0"} : (tensor<!tf_type.resource<tensor<i32>>>, tensor<i32>) -> () // CHECK-NEXT: [[r:%.*]] = "tf.ReadVariableOp"([[v]]#[[handle_idx]]) {device = "/CPU:0", dtype = i32} : (tensor<!tf_type.resource<tensor<i32>>>) -> tensor<i32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 01 23:54:14 UTC 2024 - 18.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/replicate_invariant_op_hoisting.mlir
%0:6 = tf_device.replicate([%arg0, %arg1] as %ri: tensor<*x!tf_type.resource>) {n = 2: i32} { %1 = "tf.ReadVariableOp"(%ri) {dtype = "tfdtype$DT_FLOAT"} : (tensor<*x!tf_type.resource>) -> tensor<*xf32> %2 = "tf.Shape"(%1) {device = "", T = "tfdtype$DT_FLOAT", out_type = "tfdtype$DT_INT32"} : (tensor<*xf32>) -> tensor<?xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:59:10 UTC 2023 - 11.9K bytes - Viewed (0) -
tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass.cc
std::vector<int64_t> size_as_vector; }; std::vector<int64_t> IntTensorAsVector(const Tensor& t) { DCHECK(t.dtype() == DT_INT32 || t.dtype() == DT_INT64); std::vector<int64_t> result; result.reserve(t.NumElements()); for (int i = 0; i < t.NumElements(); i++) { int64_t element = t.dtype() == DT_INT32 ? static_cast<int64_t>(t.flat<int32>()(i)) : t.flat<int64_t>()(i);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 13.8K 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) -
tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc
AttrValueMap* values) { AttrValue value; auto& type_list = *value.mutable_list(); for (auto type : types) { DataType dtype; TF_RETURN_IF_ERROR(ConvertScalarTypeToDataType(type, &dtype)); type_list.add_type(dtype); } auto result = values->insert({string(name), value}); assert(result.second && "cannot have multiple attributes with the same name"); (void)result;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device_context.cc
shape_determination_fns_.layout_preference_fn( device_tensor->shape(), device_tensor->dtype(), std::nullopt); Status status = [&]() -> Status { TF_ASSIGN_OR_RETURN(xla::Shape shape, shape_determination_fns_.shape_representation_fn( device_tensor->shape(), device_tensor->dtype(), /*fast_mem=*/false, layout_preference));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 12.7K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
operation->SetOpName(absl::StrCat("_input_", inputs_.size()).c_str())); TF_RETURN_IF_ERROR(operation->SetAttrType("dtype", dtype)); if (!shape.unknown_rank()) { TF_RETURN_IF_ERROR(operation->SetAttrShape( "shape", reinterpret_cast<int64_t*>(shape.dim_sizes().data()), shape.dims())); } int num_outputs = 1; std::vector<AbstractTensorHandle*> outputs(num_outputs);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 20:00:09 UTC 2024 - 15.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/c/eager/c_api_distributed_test.cc
" attr {" " key: 'dtype'" " value {" " type: DT_FLOAT" " }" " }" " }" " node_def {" " name: 'read1'" " op: 'ReadVariableOp'" " input: 'var'" " device: '/job:localhost/replica:0/task:1/device:CPU:0'" " attr {" " key: 'dtype'" " value {"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 09:49:45 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/cc/experimental/libtf/value.h
/// @return The unwrapped value. If this `TaggedValue` type does not currently /// contain a value of type `T`, the program terminates via a call to /// `assert`. template <typename T> T& get() { assert(type_ == EnumValueOf<T>::value); return UnionAccess<T>::unsafe_reference(*this); } /// @brief Get the underlying value based on type. /// /// @tparam T The desired return type.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 05:23:45 UTC 2024 - 20.4K bytes - Viewed (0)