- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 46 for Dtype (0.04 sec)
-
tensorflow/cc/experimental/libtf/function.cc
} Status VerifySupportedSignature(TaggedValue signature) { if (signature.type() == TaggedValue::Type::TENSOR_SPEC) { return ::tensorflow::OkStatus(); } if (signature.type() == TaggedValue::Type::TUPLE) { for (const auto& t : signature.tuple()) { if (t.type() != TaggedValue::Type::TENSOR_SPEC) { break; } } return ::tensorflow::OkStatus(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 04 19:49:06 UTC 2024 - 9.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/legalize_tf_quant_test.cc
// Set up an arg per arg_shape with the specified type. for (int i = 0; i < arg_shapes.size(); ++i) { auto metadata_arg = metadata_proto.add_args(); metadata_arg->set_kind( tensorflow::tpu::TPUCompileMetadataProto::Arg::PARAMETER); metadata_arg->set_dtype(dtype); } // Set up one dummy retval. metadata_proto.add_retvals();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 18:43:55 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/c/python_api.cc
auto* out_shape_and_type = handle_data.add_shape_and_type(); ic->ShapeHandleToProto(p.shape, out_shape_and_type->mutable_shape()); out_shape_and_type->set_dtype(p.dtype); *out_shape_and_type->mutable_type() = p.type; } } string result; handle_data.SerializeToString(&result); return result; } void SetHandleShapeAndType(TF_Graph* graph, TF_Output output, const void* proto,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jul 12 18:48:56 UTC 2023 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_tensor.cc
} else { return se::DeviceMemoryBase(const_cast<char*>(tensor.tensor_data().data()), tensor.tensor_data().size()); } } Status XlaTensor::AllocateShapedBuffer(DataType dtype, const xla::Shape& on_device_shape, xla::LocalClient* client, int device_ordinal) { xla::Shape on_host_shape =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/c/kernels/histogram_summary_op.cc
v->set_tag(tag.data(), tag.size()); histo.EncodeToProto(v->mutable_histo(), false /* Drop zero buckets */); Safe_TF_TensorPtr summary_tensor(TF_AllocateOutput( /*context=*/ctx, /*index=*/0, /*dtype=*/TF_ExpectedOutputDataType(ctx, 0), /*dims=*/nullptr, /*num_dims=*/0, /*len=*/sizeof(tensorflow::tstring), status.get())); if (TF_GetCode(status.get()) != TF_OK) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 6.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util_test.cc
auto build_result = BuildHloFromGraph(*graph, /*use_output_shapes=*/true); ASSERT_FALSE(build_result.ok()); EXPECT_THAT(build_result.message(), HasSubstr("op operand type 'tensor<2x3x4x5xi32>' and result type " "'tensor<1xi32>' are cast incompatible")); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 25 19:54:38 UTC 2024 - 9.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lift_variables.cc
// If the arg type already matches the global_tensor type, we don't need // to do anything. if (!underlying_type.empty() && underlying_type[0] == global_tensor.getType()) { assert(underlying_type.size() == 1); continue; } // Otherwise, set this argument's type to the global_tensor's type. auto new_arg_type = mlir::RankedTensorType::get(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/initialize_variables_in_session_init.cc
for (auto var_and_tensor : llvm::zip(var_ops, resource_tensors_or.value())) { auto& var_op = std::get<0>(var_and_tensor); auto& resource_tensor = std::get<1>(var_and_tensor); if (resource_tensor.dtype() != tensorflow::DT_RESOURCE) { InitializeVariable(var_op, &resource_tensor, session_init_func, builder); continue; } auto handle = resource_tensor.scalar<tensorflow::ResourceHandle>()();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/compiler/jit/variable_info_util.cc
} return absl::OkStatus(); } std::vector<int> GetResourceVariableIndicesFromContext(OpKernelContext* ctx) { std::vector<int> out; for (int64 i = 0; i < ctx->num_inputs(); i++) { if (ctx->input(i).dtype() == DT_RESOURCE) { out.push_back(i); } } return out; } Status CreateVariableInfoLookup( absl::Span<VariableInfo const> variable_args,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/cc/saved_model/bundle_v2.cc
"SavedModel checkpoint does not contain object graph."); if (object_graph_tensor.dtype() != DT_STRING || object_graph_tensor.dims() != 0 || object_graph_tensor.NumElements() != 1) { return absl::Status( absl::StatusCode::kFailedPrecondition, "SavedModel checkpoint object graph was not the correct type."); } const tstring* object_graph_string = reinterpret_cast<const tstring*>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 05 18:28:37 UTC 2023 - 9.1K bytes - Viewed (0)